add v2 for batch matmul

f0b4e8eb · Khalique · bb931fa2 · f0b4e8eb
Commit f0b4e8eb authored Aug 26, 2019 by Khalique
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

src/tf/tf.cpp src/tf/tf.cpp +1 -0

No files found.
--- a/src/tf/tf.cpp
+++ b/src/tf/tf.cpp
@@ -182,6 +182,7 @@ struct tf_parser

        add_mem_op("AvgPool", &tf_parser::parse_pooling);
        add_mem_op("BatchMatMul", &tf_parser::parse_matmul, false);
+        add_mem_op("BatchMatMulV2", &tf_parser::parse_matmul, false);
        add_mem_op("BiasAdd", &tf_parser::parse_biasadd);
        add_mem_op("Cast", &tf_parser::parse_cast, false);
        add_mem_op("ConcatV2", &tf_parser::parse_concat, false);