Home
last modified time | relevance | path

Searched refs:REDUCE_OP_SUM (Results 1 – 10 of 10) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/
Dreduce_method_info.h121 reduce_method_ = REDUCE_OP_SUM; in ReduceSumInfo()
Dunique_info.cc145 Attr attr_op = std::make_pair(OP, MakeValue(REDUCE_OP_SUM)); in ComputeReplaceGraph()
Dmatmul_info.cc238 op = CreateReduceScatterOp(REDUCE_OP_SUM, group_list[0].name()); in InferForwardCommunication()
240 op = CreateAllReduceOp(REDUCE_OP_SUM, group_list[0].name()); in InferForwardCommunication()
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()
Dgather_v2_p_info.cc738 Attr attr_op = std::make_pair(OP, MakeValue(REDUCE_OP_SUM)); in InferForwardCommunication()
792 Attr attr_op = std::make_pair(OP, MakeValue(REDUCE_OP_SUM)); in ComputeReplaceGraph()
Dops_utils.h94 constexpr char REDUCE_OP_SUM[] = "sum"; variable
Dreduce_method_info.cc204 Operator op0 = CreateAllReduceOp(REDUCE_OP_SUM, forward_group[0].name()); in CreateReduceMeanForwardOp()
Dconv2d_info.cc524 Operator op = CreateAllReduceOp(REDUCE_OP_SUM, group_list[0].name()); in InferForwardCommunication()
/third_party/mindspore/tests/ut/cpp/parallel/
Dstep_parallel_test.cc277 ValuePtr attr0_value = MakeValue(REDUCE_OP_SUM); in TEST_F()
394 ValuePtr attr0_value = MakeValue(REDUCE_OP_SUM); in TEST_F()