Searched defs:all_reduce_id (Results 1 – 9 of 9) sorted by relevance
293 auto all_reduce_id = it.first; in KeepProvablyEqualInstructionGroups() local326 auto all_reduce_id = all_reduce->all_reduce_id(); in RewriteGraph() local
120 auto all_reduce_id = hlo->all_reduce_id(); in Build() local
174 int64 all_reduce_id = 45; field
379 absl::optional<int64> all_reduce_id; in CreateFromProto() local836 const absl::optional<int64>& all_reduce_id) { in CreateAllReduce()3429 absl::optional<int64> HloInstruction::all_reduce_id() const { in all_reduce_id() function in xla::HloInstruction3434 const absl::optional<int64>& all_reduce_id) { in set_all_reduce_id()
325 absl::optional<int64> all_reduce_id() const { return all_reduce_id_; } in all_reduce_id() function
406 auto all_reduce_id = instruction->all_reduce_id(); in ComputeChannelDependencies() local
512 int64 all_reduce_id = instruction->all_reduce_id().value(); in AddMandatoryConstraints() local
522 const absl::optional<int64>& all_reduce_id) in HloAllReduceInstruction()531 const absl::optional<int64>& all_reduce_id) { in set_all_reduce_id()
822 optional<int64> all_reduce_id; in ParseInstructionRhs() local