Commit b2348b8e authored by A. Unique TensorFlower's avatar A. Unique TensorFlower Committed by TensorFlower Gardener
Browse files

Set sharding on the _Arg and _Retval nodes of a function when compiled. 

In functionalize_control_flow, set the device on the Identity node for each
value that comes out of a Switch.

PiperOrigin-RevId: 174337984
parent f6f4685b
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