[Grappler] Fix a bug with multiple-output nodes.
TrySimplifyAndReshapeUses should return a tensor not a node. Added a regression test that would have failed without this CL. ArithmeticOptimizer would have redirected the second input of concat to Split rather than Split:1. PiperOrigin-RevId: 171220303
Loading
Please sign in to comment