experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add SelectV2 to list of format agnostic ops

Browse Source
This commit is contained in:
Lukas Geiger 2020-10-16 12:11:05 +01:00
parent 716e817347
commit de3eb8516b
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tensorflow/core/grappler/optimizers/layout_optimizer.cc
View File

@ -181,6 +181,7 @@ std::set<string> GetOpsFormatAgnostic() {
"ReluGrad", "ReluGrad",
"Rint", "Rint",
"Select", "Select",
"SelectV2",
"Selu", "Selu",
"SeluGrad", "SeluGrad",
"Shape", "Shape",