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

Split out HloAllReduceInstruction as a subclass of HloInstruction. 

HloAllReduceInstruction can't subclass HloSendRecvInstruction because channel_id was optional in all reduce. So add 'all_reduce_id' instead.

PiperOrigin-RevId: 200653920
parent 332c4d69
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