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",