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

[Grappler] Fix a bug in folding multiply into add arithmetic optimization

Browse Source 
PiperOrigin-RevId: 308348068
Change-Id: I61ca637344519f9effee313ea3f911c462d9bfe6
This commit is contained in:
Eugene Zhulenev 2020-04-24 16:24:02 -07:00 committed by TensorFlower Gardener
parent 687a2d4bc5
commit 3b1545a555
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/core/grappler/optimizers/arithmetic_optimizer.cc
View File

@ -2178,6 +2178,8 @@ class FoldMultiplyIntoConv : public ArithmeticOptimizerStage {
// Check that value preserving chain is the only consumer of the Mul output.
TF_RETURN_IF_TRUE(!IsAnyMul(*source));
TF_RETURN_IF_TRUE(NumNonControlOutputs(*source, *ctx().node_map) != 1);
// And that Mul is not in the preserve set.
TF_RETURN_IF_TRUE(IsInPreserveSet(*source));
const NodeDef* mul = source;
int input_idx = 0;