From de3eb8516b161b7215b18d68479b928bcec738d6 Mon Sep 17 00:00:00 2001
From: Lukas Geiger <lukas.geiger94@gmail.com>
Date: Fri, 16 Oct 2020 12:11:05 +0100
Subject: [PATCH] Add SelectV2 to list of format agnostic ops
---
tensorflow/core/grappler/optimizers/layout_optimizer.cc | 1 +
1 file changed, 1 insertion(+)
diff --git a/tensorflow/core/grappler/optimizers/layout_optimizer.cc b/tensorflow/core/grappler/optimizers/layout_optimizer.cc
index 0bd683b7b64..2c1566d39c6 100644
--- a/tensorflow/core/grappler/optimizers/layout_optimizer.cc
+++ b/tensorflow/core/grappler/optimizers/layout_optimizer.cc
@@ -181,6 +181,7 @@ std::set<string> GetOpsFormatAgnostic() {
"ReluGrad",
"Rint",
"Select",
+ "SelectV2",
"Selu",
"SeluGrad",
"Shape",