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

Adds sigmoid to the list of operations that can be recomputed.

Browse Source 
PiperOrigin-RevId: 174047825
This commit is contained in:
A. Unique TensorFlower 2017-10-31 09:40:29 -07:00 committed by TensorFlower Gardener
parent 0d118e4dca
commit 98dad195d5
1 changed files with 5 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
tensorflow/core/grappler/optimizers/memory_optimizer.cc
View File

@ -47,12 +47,11 @@ const char* kRecomputeHint = "_recompute_hint";
// TODO(allenl): Replace this list with a cost model.
std::unordered_set<string> GetCheapToRecomputeOps() {
std::unordered_set<string> cheap_ops = {
"Add", "AddN", "BiasAdd", "Cast",
"Fill", "FloorDiv", "FloorMod", "FusedBatchNorm",
"Mul", "Neg", "RealDiv", "Reciprocal",
"Relu", "Relu6", "Reshape", "Rsqrt",
"Sqrt", "Square", "SquaredDifference", "Sub",
"Tile", "Transpose"};
"Add", "AddN", "BiasAdd", "Cast", "Fill",
"FloorDiv", "FloorMod", "FusedBatchNorm", "Mul", "Neg",
"RealDiv", "Reciprocal", "Relu", "Relu6", "Reshape",
"Rsqrt", "Sigmoid", "Sqrt", "Square", "SquaredDifference",
"Sub", "Tile", "Transpose"};
return cheap_ops;
}