Searched refs:all_reduce_id_ (Results 1 – 2 of 2) sorted by relevance
526 all_reduce_id_(all_reduce_id) { in HloAllReduceInstruction()532 all_reduce_id_ = all_reduce_id; in set_all_reduce_id()538 if (all_reduce_id_) { in ToProto()539 proto.set_all_reduce_id(*all_reduce_id_); in ToProto()561 if (all_reduce_id_) { in ExtraAttributesToStringImpl()562 result.push_back(StrCat("all_reduce_id=", *all_reduce_id_)); in ExtraAttributesToStringImpl()
325 absl::optional<int64> all_reduce_id() const { return all_reduce_id_; } in all_reduce_id()354 absl::optional<int64> all_reduce_id_; variable