Home
last modified time | relevance | path

Searched refs:CreateAllReduceOp (Results 1 – 7 of 7) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/
Dreduce_method_info.cc193 Operator op = CreateAllReduceOp(reduce_method_, forward_group[0].name()); in InferForwardCommunication()
204 Operator op0 = CreateAllReduceOp(REDUCE_OP_SUM, forward_group[0].name()); in CreateReduceMeanForwardOp()
Dtensordot_info.cc208 Operator op = CreateAllReduceOp(REDUCE_OP_SUM, forward_group[0].name()); in InferForwardCommunication()
Dunsorted_segment_op_info.cc226 Operator op = CreateAllReduceOp(REDUCE_OP_SUM, group_list[0].name()); in InferForwardCommunication()
Doperator_info.h302 Operator CreateAllReduceOp(const std::string &reduce_op, const std::string &group);
Dmatmul_info.cc240 op = CreateAllReduceOp(REDUCE_OP_SUM, group_list[0].name()); in InferForwardCommunication()
Dconv2d_info.cc524 Operator op = CreateAllReduceOp(REDUCE_OP_SUM, group_list[0].name()); in InferForwardCommunication()
Doperator_info.cc324 Operator CreateAllReduceOp(const std::string &reduce_op, const std::string &group) { in CreateAllReduceOp() function