Searched defs:all_reduce_op (Results 1 – 2 of 2) sorted by relevance
102 if (auto all_reduce_op = mlir::dyn_cast<mlir::lmhlo::AllReduceOp>(*op)) { in BefThunk() local
581 auto all_reduce_op = func_builder->create<mlir::mhlo::AllReduceOp>( in ImportInstructionImpl() local