Searched refs:set_all_reduce_barrier (Results 1 – 4 of 4) sorted by relevance
1529 void set_all_reduce_barrier(const string& barrier);
323 void set_all_reduce_barrier(string barrier) { all_reduce_barrier_ = barrier; } in set_all_reduce_barrier() function
3425 void HloInstruction::set_all_reduce_barrier(const string& barrier) { in set_all_reduce_barrier() function in xla::HloInstruction3426 return Cast<HloAllReduceInstruction>(this)->set_all_reduce_barrier(barrier); in set_all_reduce_barrier()
541 proto.set_all_reduce_barrier(all_reduce_barrier_); in ToProto()