Searched defs:all_reduces (Results 1 – 5 of 5) sorted by relevance
267 std::vector<mlir::TF::DTensorAllReduceOp> all_reduces) { in DrawAllReduceDependencies()355 const std::vector<mlir::TF::DTensorAllReduceOp>& all_reduces) { in CombineAllReduceOpsOfSameTypeAndGroupAssignment()510 const std::vector<mlir::TF::DTensorAllReduceOp>& all_reduces) { in CombineAllReduceOpsOfSameType()
737 llvm::SmallVector<mlir::TF::DTensorAllReduceOp, 4> all_reduces; in runOnOperation() local759 llvm::SmallVector<mlir::TF::DTensorReduceScatterOp, 4> all_reduces; in runOnOperation() local
98 std::vector<HloAllReduceInstruction*> all_reduces; in Run() local
64 std::vector<HloInstruction*> all_reduces; in MakeCrossReplicaReductions() local
286 std::vector<HloAllReduceInstruction*> all_reduces; in Run() local