Searched refs:all_reduce_barrier (Results 1 – 5 of 5) sorted by relevance
558 if (!all_reduce_barrier().empty()) { in ExtraAttributesToStringImpl()559 result.push_back(StrCat("barrier=\"", all_reduce_barrier(), "\"")); in ExtraAttributesToStringImpl()574 all_reduce_barrier() == casted_other.all_reduce_barrier() && in IdenticalSlowPath()583 shape, new_operands, to_apply(), replica_groups(), all_reduce_barrier(), in CloneWithNewOperandsImpl()
175 string all_reduce_barrier = 46; field
388 /*barrier=*/proto.all_reduce_barrier(), in CreateFromProto()3421 string HloInstruction::all_reduce_barrier() const { in all_reduce_barrier() function in xla::HloInstruction3422 return Cast<HloAllReduceInstruction>(this)->all_reduce_barrier(); in all_reduce_barrier()
1528 string all_reduce_barrier() const;
322 string all_reduce_barrier() const { return all_reduce_barrier_; } in all_reduce_barrier() function