OSSForks/tensorflow
1
0
Fork 0
mirror of https://github.com/zebrajr/tensorflow.git synced 2025-12-07 00:20:20 +01:00
Code Issues Actions 5 Packages 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. // TODO(allenl): Replace this list with a cost model.
std::unordered_set<string> GetCheapToRecomputeOps() { std::unordered_set<string> GetCheapToRecomputeOps() {
std::unordered_set<string> cheap_ops = { std::unordered_set<string> cheap_ops = {
"Add", "AddN", "BiasAdd", "Cast", "Add", "AddN", "BiasAdd", "Cast", "Fill",
"Fill", "FloorDiv", "FloorMod", "FusedBatchNorm", "FloorDiv", "FloorMod", "FusedBatchNorm", "Mul", "Neg",
"Mul", "Neg", "RealDiv", "Reciprocal", "RealDiv", "Reciprocal", "Relu", "Relu6", "Reshape",
"Relu", "Relu6", "Reshape", "Rsqrt", "Rsqrt", "Sigmoid", "Sqrt", "Square", "SquaredDifference",
"Sqrt", "Square", "SquaredDifference", "Sub", "Sub", "Tile", "Transpose"};
"Tile", "Transpose"};
return cheap_ops; return cheap_ops;
} }