From 1baa1bb06586bdfdc9f985bb757d20fce7213560 Mon Sep 17 00:00:00 2001
From: Yu-Cheng Ling <ycling@google.com>
Date: Tue, 23 Jul 2019 11:37:18 -0700
Subject: [PATCH] Fix Toco Flex tests for newly added ops

PiperOrigin-RevId: 259578274
---
 tensorflow/lite/delegates/flex/whitelisted_flex_ops.cc | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/tensorflow/lite/delegates/flex/whitelisted_flex_ops.cc b/tensorflow/lite/delegates/flex/whitelisted_flex_ops.cc
index 4ade603ce2f..2459bd157f6 100644
--- 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",