Add SelectV2 to list of format agnostic ops

2020-10-16 12:11:05 +01:00 · 2020-10-16 12:11:05 +01:00 · de3eb8516b
commit de3eb8516b
parent 716e817347
1 changed files with 1 additions and 0 deletions
--- 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",