Skip to content

Commit aaab800

Browse files
authored
Add mapping for tf lite op TFL_BATCH_MATMUL and TFL_MATMUL. (#1950)
Signed-off-by: Jay Zhang <jiz@microsoft.com>
1 parent e099356 commit aaab800

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

tf2onnx/tflite_handlers/tfl_direct.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,7 @@
1212

1313

1414
@tfl_op("TFL_ABS", tf_op="Abs")
15+
@tfl_op("TFL_BATCH_MATMUL", tf_op="BatchMatMul")
1516
@tfl_op("TFL_BROADCAST_TO", tf_op="BroadcastTo")
1617
@tfl_op("TFL_CEIL", tf_op="Ceil")
1718
@tfl_op("TFL_COS", tf_op="Cos")
@@ -30,6 +31,7 @@
3031
@tfl_op("TFL_LOGICAL_AND", tf_op="LogicalAnd")
3132
@tfl_op("TFL_LOGICAL_NOT", tf_op="LogicalNot")
3233
@tfl_op("TFL_LOGICAL_OR", tf_op="LogicalOr")
34+
@tfl_op("TFL_MATMUL", tf_op="MatMul")
3335
@tfl_op("TFL_MATRIX_DIAG", tf_op="MatrixDiag")
3436
@tfl_op("TFL_MATRIX_SET_DIAG", tf_op="MatrixSetDiag")
3537
@tfl_op("TFL_MAXIMUM", tf_op="Maximum")

0 commit comments

Comments
 (0)