Fix Toco Flex tests for newly added ops

PiperOrigin-RevId: 259578274
2019-07-23 11:37:18 -07:00 · 2019-07-23 11:37:18 -07:00 · 1baa1bb065
commit 1baa1bb065
parent bc46584917
1 changed files with 2 additions and 0 deletions
--- a/tensorflow/lite/delegates/flex/whitelisted_flex_ops.cc
+++ b/tensorflow/lite/delegates/flex/whitelisted_flex_ops.cc
@ -182,7 +182,9 @@ bool IsWhitelistedFlexOp(const std::string& tensorflow_op_name) {
          "LRN",
          "MatMul",
          "MatrixDiag",
+          "MatrixDiagV2",
          "MatrixSetDiag",
+          "MatrixSetDiagV2",
          "Max",
          "Maximum",
          "MaxPool",