Skip to content

Commit c328ded

Browse files
Pey-cryptorkazants
andauthored
[TF FE] Implementation of BatchMatrixInverse Support (openvinotoolkit#26033)
Implemented BatchMatrixInverse Registered into OP_loader Implemented associated pytests For Issue openvinotoolkit#24807 --------- Co-authored-by: Roman Kazantsev <roman.kazantsev@intel.com>
1 parent 00d9800 commit c328ded

File tree

3 files changed

+3
-2
lines changed

3 files changed

+3
-2
lines changed

src/frontends/tensorflow/docs/supported_ops.md

+1-1
Original file line numberDiff line numberDiff line change
@@ -111,7 +111,7 @@ A "supported operation" is one that TensorFlow Frontend can convert to the OpenV
111111
| BatchMatrixDeterminant | NO | |
112112
| BatchMatrixDiag | NO | |
113113
| BatchMatrixDiagPart | NO | |
114-
| BatchMatrixInverse | NO | |
114+
| BatchMatrixInverse | YES | |
115115
| BatchMatrixSetDiag | NO | |
116116
| BatchMatrixSolve | NO | |
117117
| BatchMatrixSolveLs | NO | |

src/frontends/tensorflow/src/op_table.cpp

+1
Original file line numberDiff line numberDiff line change
@@ -221,6 +221,7 @@ const std::map<std::string, CreatorFunction> get_supported_ops() {
221221
{"BatchMatMul", CreatorFunction(translate_batch_mat_mul_op)},
222222
{"BatchMatMulV2", CreatorFunction(translate_batch_mat_mul_op)},
223223
{"BatchMatMulV3", CreatorFunction(translate_batch_mat_mul_with_type_op)},
224+
{"BatchMatrixInverse", CreatorFunction(translate_matrix_inverse_op)},
224225
{"BatchToSpaceND", CreatorFunction(translate_batch_to_space_nd_op)},
225226
{"BroadcastArgs", CreatorFunction(translate_broadcast_args_op)},
226227
{"BroadcastTo", CreatorFunction(translate_broadcast_to_op)},

src/frontends/tensorflow_common/src/op/matrix_inverse.cpp

+1-1
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@ namespace frontend {
1313
namespace tensorflow {
1414
namespace op {
1515
OutputVector translate_matrix_inverse_op(const NodeContext& node) {
16-
default_op_checks(node, 1, {"MatrixInverse"});
16+
default_op_checks(node, 1, {"MatrixInverse", "BatchMatrixInverse"});
1717
// retrieve the input tensor
1818
auto input = node.get_input(0);
1919

0 commit comments

Comments
 (0)