From 58a9e935bbc1c72e2a3e4e833ab60f78e2b3afee Mon Sep 17 00:00:00 2001 From: Tomiwa Philip <110630370+TomiwaPhilip@users.noreply.github.com> Date: Wed, 16 Aug 2023 03:36:52 +0000 Subject: [PATCH] Added lu_solve function in linalg --- ivy/functional/frontends/tensorflow/linalg.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/ivy/functional/frontends/tensorflow/linalg.py b/ivy/functional/frontends/tensorflow/linalg.py index 76c768cd4d704..bff777c5db341 100644 --- a/ivy/functional/frontends/tensorflow/linalg.py +++ b/ivy/functional/frontends/tensorflow/linalg.py @@ -470,3 +470,8 @@ def tensor_diag_part(input, name=None): reshaped = ivy.reshape(input, (prod, prod)) diagonal = ivy.diagonal(reshaped) return ivy.reshape(diagonal, tuple(half_shape)) + + +@to_ivy_arrays_and_back +def lu_solve(lower_upper, perm, rhs, validate_args=False, name=None): + return ivy.solve(lower_upper, perm, rhs, validate_args=False, name=None)