Commit d41bc208 authored by Yu-Cheng Ling's avatar Yu-Cheng Ling Committed by TensorFlower Gardener
Browse files

Fix a bug in Toco FuseBinaryIntoFollowingAffine transformation 

It incorrectly removes an array even if it has other consumers.
`GetOpWithInput` function returns the first op when there are
multiple consumer ops.

The transformation can be further optimized, but this CL is
strictly better than before.

PiperOrigin-RevId: 223850263
parent 5c25275e
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