Commit 4a0bbf36 authored by Guangda Lai's avatar Guangda Lai Committed by TensorFlower Gardener
Browse files

Fix MulConvPushDown so it applies the optimization only when the shape of the 

const input to Mul op has at most one dimension that is larger than one.

Without the change in constant_folding.cc, the updated tests will fail with
incorrect result.

PiperOrigin-RevId: 233693586
parent 4315fcbc
Show whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment