Searched refs:set_all_reduce_id (Results 1 – 6 of 6) sorted by relevance
371 next->set_all_reduce_id(all_reduce_id); in RewriteGraph()
1533 void set_all_reduce_id(const absl::optional<int64>& all_reduce_id);
530 void HloAllReduceInstruction::set_all_reduce_id( in set_all_reduce_id() function in xla::HloAllReduceInstruction539 proto.set_all_reduce_id(*all_reduce_id_); in ToProto()
326 void set_all_reduce_id(const absl::optional<int64>& all_reduce_id);
3433 void HloInstruction::set_all_reduce_id( in set_all_reduce_id() function in xla::HloInstruction3435 return Cast<HloAllReduceInstruction>(this)->set_all_reduce_id(all_reduce_id); in set_all_reduce_id()
2150 instr.set_all_reduce_id(channel_id->handle()); in CrossReplicaSum()