| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/profiling/reporter/ |
| D | task_desc_reporter.cc | 34 if (AnfAlgo::GetKernelType(node) != TBE_KERNEL && AnfAlgo::GetKernelType(node) != AKG_KERNEL) { in ReportData()
|
| D | graph_desc_reporter.cc | 26 if (AnfAlgo::GetKernelType(node) != TBE_KERNEL && AnfAlgo::GetKernelType(node) != AKG_KERNEL) { in ReportData()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/buffer_fusion/ |
| D | fusion_base_pass.cc | 33 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()
|
| D | matmul_eltwise_fusion_pass.cc | 54 if (AnfAlgo::GetKernelType(cnode) == KernelType::TBE_KERNEL && in MatchSingleFusionPattern()
62 if (AnfAlgo::GetKernelType(cnode) == KernelType::TBE_KERNEL && in MatchSingleFusionPattern()
|
| D | conv_single_in_fusion_pass.cc | 49 if (AnfAlgo::GetKernelType(eltwise_input) == KernelType::TBE_KERNEL && in MatchConvSingleInEltwise()
68 if (AnfAlgo::GetKernelType(cnode) == KernelType::TBE_KERNEL && in MatchSingleFusionPattern()
|
| D | conv_double_in_fusion_pass.cc | 45 if (AnfAlgo::GetKernelType(double_in_eltwise_input) == KernelType::TBE_KERNEL && in MatchConvDoubleInEltwise()
64 if (AnfAlgo::GetKernelType(cnode) == KernelType::TBE_KERNEL && in MatchSingleFusionPattern()
|
| D | segment_eltwise_fusion_pass.cc | 45 if (AnfAlgo::GetKernelType(eltwise_input) == KernelType::TBE_KERNEL && in MatchSegmentEltwise()
78 if (AnfAlgo::GetKernelType(cnode) == KernelType::TBE_KERNEL && in MatchSingleFusionPattern()
|
| D | reduce_eltwise_fusion_pass.cc | 47 if (AnfAlgo::GetKernelType(eltwise_input) == KernelType::TBE_KERNEL && in MatchReduceEltwise()
81 if (AnfAlgo::GetKernelType(cnode) == KernelType::TBE_KERNEL && in MatchSingleFusionPattern()
|
| D | bnupdate_eltwise_fusion_pass.cc | 58 if (AnfAlgo::GetKernelType(cnode) == KernelType::TBE_KERNEL && in MatchSingleFusionPattern()
|
| D | eltwise_fusion_pass.cc | 67 if (AnfAlgo::GetKernelType(cnode) == KernelType::TBE_KERNEL && in MatchSingleFusionPattern()
|
| D | multi_output_fusion_pass.cc | 72 if (AnfAlgo::GetKernelType(cnode) == KernelType::TBE_KERNEL && in MatchSingleFusionPattern()
|
| D | depthwiseconv_eltwise_fusion_pass.cc | 63 if (AnfAlgo::GetKernelType(cnode) == KernelType::TBE_KERNEL && in MatchSingleFusionPattern()
|
| D | bnupdate_eltwise_eltwise_fusion_pass.cc | 69 if (AnfAlgo::GetKernelType(cnode) == KernelType::TBE_KERNEL && in MatchSingleFusionPattern()
|
| D | stridedread_conv_stridedwrite_fusion_pass.cc | 45 if (AnfAlgo::GetKernelType(conv_cnode) == KernelType::TBE_KERNEL && in MatchStridedReadConvStridedWrite()
|
| D | conv2dbackprop_eltwise_eltwise_fusion_pass.cc | 80 if (AnfAlgo::GetKernelType(cnode) == KernelType::TBE_KERNEL && in MatchSingleFusionPattern()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ |
| D | kernel_build_ascend.cc | 48 KernelType kernel_type = AnfAlgo::GetKernelType(anf_node); in SerialCompileImpl()
83 KernelType kernel_type = AnfAlgo::GetKernelType(anf_node); in KernelBuildParallelCompile()
368 AnfAlgo::GetKernelType(anf_node) == KernelType::AKG_KERNEL) { in TbeClearZeroNodeFusion()
388 … (AnfAlgo::GetKernelType(anf_node) == KernelType::TBE_KERNEL && IsAtomicNode(anf_node))) { in TbeClearZeroNodeFusion()
439 AnfAlgo::GetKernelType(anf_node) == KernelType::AKG_KERNEL) { in KernelBuildPreprocess()
458 } else if (AnfAlgo::GetKernelType(anf_node) == KernelType::TBE_KERNEL) { in KernelBuildPreprocess()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/executor/ |
| D | dynamic_kernel.cc | 55 int DynamicKernel::GetKernelType() const { return AnfAlgo::GetKernelType(cnode_ptr_.lock()); } in GetKernelType() function in mindspore::device::DynamicKernel
|
| D | dynamic_kernel.h | 46 [[nodiscard]] int GetKernelType() const;
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/dump/ |
| D | data_dumper.cc | 78 if (AnfAlgo::GetKernelType(kernel) == HCCL_KERNEL && in GetNeedDumpKernelList()
206 …if (AnfAlgo::GetKernelType(kernel) != TBE_KERNEL && AnfAlgo::GetKernelType(kernel) != AICPU_KERNEL… in KernelNeedDump()
207 AnfAlgo::GetKernelType(kernel) != AKG_KERNEL) { in KernelNeedDump()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/profiling/ |
| D | profiling_utils.cc | 209 … if (AnfAlgo::GetKernelType(*iter) == TBE_KERNEL || AnfAlgo::GetKernelType(*iter) == AKG_KERNEL || in GetGraphLastKernelName()
386 if (AnfAlgo::GetKernelType(node) == TBE_KERNEL || AnfAlgo::GetKernelType(node) == AKG_KERNEL) { in ValidComputeGraph()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/ |
| D | gpu_kernel_build.cc | 43 if (session::AnfRuntimeAlgorithm::GetKernelType(kernel) == KernelType::AKG_KERNEL) { in CreateGPUKernel()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/pass/ |
| D | replace_node_by_proxy.cc | 46 builder.SetKernelType(AnfAlgo::GetKernelType(cnode)); in GenerateKernelBuildInfo()
|
| /third_party/mindspore/tests/ut/cpp/session/ |
| D | anf_runtime_algorithm_test.cc | 637 TEST_F(AnfRuntimeAlgorithmTest, GetKernelType) { in TEST_F() argument
649 EXPECT_EQ(AnfAlgo::GetKernelType(add), AKG_KERNEL); in TEST_F()
650 EXPECT_THROW(AnfAlgo::GetKernelType(nullptr), std::runtime_error); in TEST_F()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/ |
| D | reorder_ops.cc | 126 info_builder.SetKernelType(AnfAlgo::GetKernelType(orig_node)); in SetNodeInfo()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/tasksink/ |
| D | task_generator.cc | 99 if (AnfAlgo::GetKernelType(anf_node_ptr) == KernelType::AKG_KERNEL) { in LaunchAddrCleanKernel()
|