Added lu_solve function in linalg

ivy-llc · Aug 16, 2023 · 58a9e93 · 58a9e93
1 parent 74c13af
commit 58a9e93
Showing 1 changed file with 5 additions and 0 deletions.
diff --git 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)