| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/ |
| D | common_utils.cc | 88 const std::unordered_map<FusionType, std::string> fusion_type_name_maps = {
89 {FusionType::BN_UPDATE_GRAD, "bn_update_grad"},
90 {FusionType::BN_GRAD_REDUCE, "bn_grad_reduce"},
91 {FusionType::LAYER_NORM_GRAD, "layer_norm_grad"},
92 {FusionType::L2LOSS_MUL_ADDN, "l2loss_mul_addn"},
93 {FusionType::ELEMWISE, "ElemWise"},
94 {FusionType::PURE_BROADCAST, "PureBroadcast"},
95 {FusionType::COMMREDUCE, "CommReduce"},
96 {FusionType::SEGMENT, "Segment"},
97 {FusionType::INPLACE, "Inplace"},
[all …]
|
| D | kernel_build_info.h | 89 FusionType fusion_type() const { return fusion_type_; } in fusion_type()
119 FusionType fusion_type_;
167 void SetFusionType(FusionType fusion_type);
|
| D | common_utils.h | 109 FusionType GetFusionTypeByName(const std::string &name);
110 std::string GetFusionNameByType(const kernel::FusionType &type);
|
| /third_party/mindspore/tests/ut/cpp/pre_activate/ascend/buffer_fusion/ |
| D | buffer_fusion_test.cc | 72 builder.SetFusionType(kernel::FusionType::ELEMWISE); in TEST_F()
87 builder1.SetFusionType(kernel::FusionType::OPAQUE); in TEST_F()
143 builder.SetFusionType(kernel::FusionType::ELEMWISE); in TEST_F()
166 builder1.SetFusionType(kernel::FusionType::OPAQUE); in TEST_F()
179 builder2.SetFusionType(kernel::FusionType::COMMREDUCE); in TEST_F()
233 builder.SetFusionType(kernel::FusionType::ELEMWISE); in TEST_F()
256 builder1.SetFusionType(kernel::FusionType::OPAQUE); in TEST_F()
269 builder2.SetFusionType(kernel::FusionType::COMMREDUCE); in TEST_F()
315 builder.SetFusionType(kernel::FusionType::ELEMWISE); in TEST_F()
327 builder2.SetFusionType(kernel::FusionType::OPAQUE); in TEST_F()
[all …]
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/buffer_fusion/ |
| D | fusion_base_pass.cc | 34 …AnfAlgo::GetFusionType(node) == kernel::FusionType::ELEMWISE && not_updatestate_nums == ELTWISE_US… in CheckEltWiseNode()
49 …AnfAlgo::GetFusionType(node) == kernel::FusionType::ELEMWISE && not_updatestate_nums == ELTWISE_US… in CheckDoubleInEltWiseNode()
64 …AnfAlgo::GetFusionType(node) == kernel::FusionType::ELEMWISE && not_updatestate_nums == ELTWISE_MU… in CheckMultiOutputEltWiseNode()
|
| D | matmul_eltwise_fusion_pass.cc | 55 AnfAlgo::GetFusionType(cnode) == kernel::FusionType::ELEMWISE && in MatchSingleFusionPattern()
63 AnfAlgo::GetFusionType(cnode) == kernel::FusionType::ELEMWISE && in MatchSingleFusionPattern()
|
| D | conv_single_in_fusion_pass.cc | 50 AnfAlgo::GetFusionType(eltwise_input) == kernel::FusionType::CONV) { in MatchConvSingleInEltwise()
69 …AnfAlgo::GetFusionType(cnode) == kernel::FusionType::ELEMWISE && cnode->inputs().size() == ELTWISE… in MatchSingleFusionPattern()
|
| D | conv_double_in_fusion_pass.cc | 46 AnfAlgo::GetFusionType(double_in_eltwise_input) == kernel::FusionType::CONV) { in MatchConvDoubleInEltwise()
65 …AnfAlgo::GetFusionType(cnode) == kernel::FusionType::ELEMWISE && cnode->inputs().size() == ELTWISE… in MatchSingleFusionPattern()
|
| D | segment_eltwise_fusion_pass.cc | 46 AnfAlgo::GetFusionType(eltwise_input) == kernel::FusionType::SEGMENT) { in MatchSegmentEltwise()
79 …AnfAlgo::GetFusionType(cnode) == kernel::FusionType::ELEMWISE && cnode->inputs().size() == ELTWISE… in MatchSingleFusionPattern()
|
| D | reduce_eltwise_fusion_pass.cc | 48 AnfAlgo::GetFusionType(eltwise_input) == kernel::FusionType::COMMREDUCE && in MatchReduceEltwise()
82 …AnfAlgo::GetFusionType(cnode) == kernel::FusionType::ELEMWISE && cnode->inputs().size() == ELTWISE… in MatchSingleFusionPattern()
|
| D | bnupdate_eltwise_fusion_pass.cc | 59 AnfAlgo::GetFusionType(cnode) == kernel::FusionType::ELEMWISE && in MatchSingleFusionPattern()
|
| D | eltwise_fusion_pass.cc | 68 …AnfAlgo::GetFusionType(cnode) == kernel::FusionType::ELEMWISE && cnode->inputs().size() == ELTWISE… in MatchSingleFusionPattern()
|
| D | multi_output_fusion_pass.cc | 73 …AnfAlgo::GetFusionType(cnode) == kernel::FusionType::ELEMWISE && cnode->inputs().size() == ELTWISE… in MatchSingleFusionPattern()
|
| D | depthwiseconv_eltwise_fusion_pass.cc | 64 AnfAlgo::GetFusionType(cnode) == kernel::FusionType::ELEMWISE) { in MatchSingleFusionPattern()
|
| D | bnupdate_eltwise_eltwise_fusion_pass.cc | 70 AnfAlgo::GetFusionType(cnode) == kernel::FusionType::ELEMWISE && in MatchSingleFusionPattern()
|
| D | stridedread_conv_stridedwrite_fusion_pass.cc | 46 AnfAlgo::GetFusionType(conv_cnode) == kernel::FusionType::CONV && in MatchStridedReadConvStridedWrite()
|
| D | conv2dbackprop_eltwise_eltwise_fusion_pass.cc | 81 AnfAlgo::GetFusionType(cnode) == kernel::FusionType::ELEMWISE && in MatchSingleFusionPattern()
|
| /third_party/mindspore/tests/ut/cpp/pre_activate/ascend/format_type/ |
| D | insert_cast_test.cc | 61 builder.SetFusionType(kernel::FusionType::ELEMWISE); in TEST_F()
69 builder1.SetFusionType(kernel::FusionType::ELEMWISE); in TEST_F()
123 builder1.SetFusionType(kernel::FusionType::ELEMWISE); in TEST_F()
|
| /third_party/mindspore/tests/ut/cpp/pre_activate/pass/ |
| D | allreduce_fusion_test.cc | 57 builder.SetFusionType(kernel::FusionType::ELEMWISE); in TEST_F()
98 builder.SetFusionType(kernel::FusionType::ELEMWISE); in TEST_F()
139 builder.SetFusionType(kernel::FusionType::ELEMWISE); in TEST_F()
196 builder.SetFusionType(kernel::FusionType::ELEMWISE); in TEST_F()
|
| /third_party/mindspore/tests/ut/cpp/tbe/ |
| D | tbe_json_creator_test.cc | 67 builder.SetFusionType(kernel::FusionType::ELEMWISE); in TEST_F()
110 builder.SetFusionType(kernel::FusionType::ELEMWISE); in TEST_F()
168 builder.SetFusionType(kernel::FusionType::ELEMWISE); in TEST_F()
222 builder.SetFusionType(kernel::FusionType::ELEMWISE); in TEST_F()
267 builder.SetFusionType(kernel::FusionType::ELEMWISE); in TEST_F()
282 builder1.SetFusionType(kernel::FusionType::OPAQUE); in TEST_F()
339 builder.SetFusionType(kernel::FusionType::ELEMWISE); in TEST_F()
|
| /third_party/mindspore/tests/ut/cpp/pre_activate/ascend/ir_fission/ |
| D | transdata_split_test.cc | 126 builder.SetFusionType(kernel::FusionType::ELEMWISE); in TEST_F()
174 builder.SetFusionType(kernel::FusionType::ELEMWISE); in TEST_F()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/rts/ |
| D | rt_kernel_info.cc | 82 kernel_build_info_builder->SetFusionType(kernel::FusionType::OPAQUE); in GetRtKelInfo()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/ |
| D | parallel_cost_model.h | 64 std::string FusionType() const { return fusion_type_; } in FusionType() function
|
| D | decrease_compute_precision.cc | 47 builder.SetFusionType(kernel::FusionType::OPAQUE); in AddCastCNode()
188 graph_info_builder.SetFusionType(kernel::FusionType::OPAQUE); in Process()
|
| D | decrease_transfer_precision.cc | 173 graph_info_builder.SetFusionType(kernel::FusionType::OPAQUE); in Process_Father()
286 node_info_builder.SetFusionType(kernel::FusionType::OPAQUE); in Process_Son()
|