Searched defs:set_all_reduce_barrier (Results 1 – 2 of 2) sorted by relevance
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::HloInstruction