Searched refs:all_reduce_op (Results 1 – 2 of 2) sorted by relevance
1077 TF_ASSIGN_OR_RETURN(auto all_reduce_op, in EmitAllReduceOp()1081 SetupCommonCollectiveOpAttributes(all_reduce_op, instr, builder_)); in EmitAllReduceOp()1082 all_reduce_op.use_global_device_idsAttr( in EmitAllReduceOp()1085 *instr->called_computations()[0], &all_reduce_op.computation(), in EmitAllReduceOp()1087 return all_reduce_op; in EmitAllReduceOp()
559 auto all_reduce_op = func_builder->create<mlir::mhlo::AllReduceOp>( in ImportInstructionImpl() local562 &all_reduce_op.computation())); in ImportInstructionImpl()563 return all_reduce_op.getOperation(); in ImportInstructionImpl()