Commit 1899ac3d authored by Adrian Kuegel's avatar Adrian Kuegel Committed by TensorFlower Gardener
Browse files

Make use of the Iota function that gets passed a shape. 

This avoids manual broadcasts which are handled automatically by the Iota op.

PiperOrigin-RevId: 222949991
parent 4fbbeea2
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