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

Replace NcclReduce/Broadcast ops during graph optimization so that we can... 

Replace NcclReduce/Broadcast ops during graph optimization so that we can generate gradients for Reduce/Broadcast.
Changing _NcclBroadcastRecv shape input to int32 so that the corresponding Const op is outputting to HostMem.

PiperOrigin-RevId: 172279684
parent e30246c4
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