Home
last modified time | relevance | path

Searched refs:TBE_KERNEL (Results 1 – 25 of 41) sorted by relevance

12

/third_party/mindspore/tests/ut/cpp/pre_activate/ascend/buffer_fusion/
Dbuffer_fusion_test.cc71 builder.SetKernelType(KernelType::TBE_KERNEL); in TEST_F()
74 builder.SetKernelType(KernelType::TBE_KERNEL); in TEST_F()
86 builder1.SetKernelType(KernelType::TBE_KERNEL); in TEST_F()
89 builder1.SetKernelType(KernelType::TBE_KERNEL); in TEST_F()
142 builder.SetKernelType(KernelType::TBE_KERNEL); in TEST_F()
145 builder.SetKernelType(KernelType::TBE_KERNEL); in TEST_F()
165 builder1.SetKernelType(KernelType::TBE_KERNEL); in TEST_F()
168 builder1.SetKernelType(KernelType::TBE_KERNEL); in TEST_F()
178 builder2.SetKernelType(KernelType::TBE_KERNEL); in TEST_F()
181 builder2.SetKernelType(KernelType::TBE_KERNEL); in TEST_F()
[all …]
/third_party/mindspore/tests/ut/cpp/pre_activate/pass/
Dcommon_subexpression_elimination_test.cc117 … {kNumberTypeFloat32, kNumberTypeFloat16}, {kNumberTypeFloat32}, TBE_KERNEL), in TEST_F()
155 … {kNumberTypeFloat32, kNumberTypeFloat32}, {kNumberTypeFloat32}, TBE_KERNEL), in TEST_F()
159 … {kNumberTypeFloat32, kNumberTypeFloat16}, {kNumberTypeFloat32}, TBE_KERNEL), in TEST_F()
166 … {kNumberTypeFloat32, kNumberTypeFloat32}, {kNumberTypeFloat32}, TBE_KERNEL), in TEST_F()
170 … {kNumberTypeFloat32, kNumberTypeFloat32}, {kNumberTypeFloat32}, TBE_KERNEL), in TEST_F()
177 … {kNumberTypeFloat32, kNumberTypeFloat32}, {kNumberTypeFloat32}, TBE_KERNEL), in TEST_F()
188 … {kNumberTypeFloat32, kNumberTypeFloat32}, {kNumberTypeFloat32}, TBE_KERNEL), in TEST_F()
192 … {kNumberTypeFloat32, kNumberTypeFloat32}, {kNumberTypeFloat32}, TBE_KERNEL), in TEST_F()
/third_party/mindspore/tests/ut/cpp/tbe/
Dtbe_json_creator_test.cc66 builder.SetKernelType(KernelType::TBE_KERNEL); in TEST_F()
69 builder.SetKernelType(KernelType::TBE_KERNEL); in TEST_F()
109 builder.SetKernelType(KernelType::TBE_KERNEL); in TEST_F()
112 builder.SetKernelType(KernelType::TBE_KERNEL); in TEST_F()
167 builder.SetKernelType(KernelType::TBE_KERNEL); in TEST_F()
170 builder.SetKernelType(KernelType::TBE_KERNEL); in TEST_F()
221 builder.SetKernelType(KernelType::TBE_KERNEL); in TEST_F()
224 builder.SetKernelType(KernelType::TBE_KERNEL); in TEST_F()
266 builder.SetKernelType(KernelType::TBE_KERNEL); in TEST_F()
269 builder.SetKernelType(KernelType::TBE_KERNEL); in TEST_F()
[all …]
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/buffer_fusion/
Dfusion_base_pass.cc33 return AnfAlgo::GetKernelType(node) == KernelType::TBE_KERNEL && in CheckEltWiseNode()
48 return AnfAlgo::GetKernelType(node) == KernelType::TBE_KERNEL && in CheckDoubleInEltWiseNode()
63 return AnfAlgo::GetKernelType(node) == KernelType::TBE_KERNEL && in CheckMultiOutputEltWiseNode()
Dmatmul_eltwise_fusion_pass.cc54 if (AnfAlgo::GetKernelType(cnode) == KernelType::TBE_KERNEL && in MatchSingleFusionPattern()
62 if (AnfAlgo::GetKernelType(cnode) == KernelType::TBE_KERNEL && in MatchSingleFusionPattern()
Dconv_single_in_fusion_pass.cc49 if (AnfAlgo::GetKernelType(eltwise_input) == KernelType::TBE_KERNEL && in MatchConvSingleInEltwise()
68 if (AnfAlgo::GetKernelType(cnode) == KernelType::TBE_KERNEL && in MatchSingleFusionPattern()
Dconv_double_in_fusion_pass.cc45 if (AnfAlgo::GetKernelType(double_in_eltwise_input) == KernelType::TBE_KERNEL && in MatchConvDoubleInEltwise()
64 if (AnfAlgo::GetKernelType(cnode) == KernelType::TBE_KERNEL && in MatchSingleFusionPattern()
Dsegment_eltwise_fusion_pass.cc45 if (AnfAlgo::GetKernelType(eltwise_input) == KernelType::TBE_KERNEL && in MatchSegmentEltwise()
78 if (AnfAlgo::GetKernelType(cnode) == KernelType::TBE_KERNEL && in MatchSingleFusionPattern()
Dreduce_eltwise_fusion_pass.cc47 if (AnfAlgo::GetKernelType(eltwise_input) == KernelType::TBE_KERNEL && in MatchReduceEltwise()
81 if (AnfAlgo::GetKernelType(cnode) == KernelType::TBE_KERNEL && in MatchSingleFusionPattern()
Dbnupdate_eltwise_fusion_pass.cc58 if (AnfAlgo::GetKernelType(cnode) == KernelType::TBE_KERNEL && in MatchSingleFusionPattern()
Deltwise_fusion_pass.cc67 if (AnfAlgo::GetKernelType(cnode) == KernelType::TBE_KERNEL && in MatchSingleFusionPattern()
Dmulti_output_fusion_pass.cc72 if (AnfAlgo::GetKernelType(cnode) == KernelType::TBE_KERNEL && in MatchSingleFusionPattern()
Ddepthwiseconv_eltwise_fusion_pass.cc63 if (AnfAlgo::GetKernelType(cnode) == KernelType::TBE_KERNEL && in MatchSingleFusionPattern()
Dstridedread_conv_stridedwrite_fusion_pass.cc45 if (AnfAlgo::GetKernelType(conv_cnode) == KernelType::TBE_KERNEL && in MatchStridedReadConvStridedWrite()
Dbnupdate_eltwise_eltwise_fusion_pass.cc69 if (AnfAlgo::GetKernelType(cnode) == KernelType::TBE_KERNEL && in MatchSingleFusionPattern()
Dconv2dbackprop_eltwise_eltwise_fusion_pass.cc80 if (AnfAlgo::GetKernelType(cnode) == KernelType::TBE_KERNEL && in MatchSingleFusionPattern()
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/
Dkernel_build_ascend.cc85 case KernelType::TBE_KERNEL: { in KernelBuildParallelCompile()
177 builder->SetKernelType(KernelType::TBE_KERNEL); in AddTbeClearZeroNode()
204 builder->SetKernelType(KernelType::TBE_KERNEL); in AddFusionTbeClearZeroNode()
388 … (AnfAlgo::GetKernelType(anf_node) == KernelType::TBE_KERNEL && IsAtomicNode(anf_node))) { in TbeClearZeroNodeFusion()
458 } else if (AnfAlgo::GetKernelType(anf_node) == KernelType::TBE_KERNEL) { in KernelBuildPreprocess()
Dascend_launch_atomic_clean.cc108 builder->SetKernelType(KernelType::TBE_KERNEL); in ConstructKernelGraphAndSetAttr()
Dascend_launch_transdata.cc104 builder->SetKernelType(KernelType::TBE_KERNEL); in ConstructKernelGraphAndSetAttr()
/third_party/mindspore/tests/ut/cpp/pre_activate/ascend/ir_fission/
Dbn_split_test.cc86 builder.SetKernelType(KernelType::TBE_KERNEL); in TEST_F()
135 builder.SetKernelType(KernelType::TBE_KERNEL); in TEST_F()
Dtransdata_split_test.cc125 builder.SetKernelType(KernelType::TBE_KERNEL); in TEST_F()
173 builder.SetKernelType(KernelType::TBE_KERNEL); in TEST_F()
Dbn_grad_split_test.cc94 builder1.SetKernelType(TBE_KERNEL); in TEST_F()
156 builder1.SetKernelType(TBE_KERNEL); in TEST_F()
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/profiling/reporter/
Dtask_desc_reporter.cc34 if (AnfAlgo::GetKernelType(node) != TBE_KERNEL && AnfAlgo::GetKernelType(node) != AKG_KERNEL) { in ReportData()
Dgraph_desc_reporter.cc26 if (AnfAlgo::GetKernelType(node) != TBE_KERNEL && AnfAlgo::GetKernelType(node) != AKG_KERNEL) { in ReportData()
/third_party/mindspore/tests/ut/cpp/pre_activate/mem_reuse/
Dmem_reuse_test.cc98 builder.SetKernelType(KernelType::TBE_KERNEL); in CreateKernelGraph()
128 relu_builder.SetKernelType(KernelType::TBE_KERNEL); in CreateKernelGraph()

12