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 GetOpsFormatAgnostic() { "ReluGrad", "Rint", "Select", + "SelectV2", "Selu", "SeluGrad", "Shape",