Searched refs:all_reduce_barrier_ (Results 1 – 2 of 2) sorted by relevance
322 string all_reduce_barrier() const { return all_reduce_barrier_; } in all_reduce_barrier()323 void set_all_reduce_barrier(string barrier) { all_reduce_barrier_ = barrier; } in set_all_reduce_barrier()349 string all_reduce_barrier_; variable
525 all_reduce_barrier_(barrier), in HloAllReduceInstruction()541 proto.set_all_reduce_barrier(all_reduce_barrier_); in ToProto()