Searched defs:all_reduce_group (Results 1 – 1 of 1) sorted by relevance
149 std::vector<mlir::TF::DTensorAllReduceOp>& all_reduce_group) { in MergeAllReduceGroup()466 for (auto& all_reduce_group : all_reduce_groups) { in CombineAllReduceOpsOfSameTypeAndGroupAssignment() local