Searched refs:KernelBuildInfoPtr (Results 1 – 25 of 36) sorted by relevance
12
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ |
| D | kernel_info.h | 47 kernel::KernelBuildInfoPtr GetMutableSelectKernelBuildInfo() const;
48 void set_select_kernel_build_info(const kernel::KernelBuildInfoPtr &select_kernel_build_info) { in set_select_kernel_build_info()
78 kernel::KernelBuildInfoPtr select_kernel_build_info_;
|
| D | kernel_info.cc | 23 kernel::KernelBuildInfoPtr KernelInfo::GetMutableSelectKernelBuildInfo() const { return select_kern… in GetMutableSelectKernelBuildInfo()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/ |
| D | kernel_query.cc | 136 bool IsSupportedByAICPU(const AnfNodePtr &kernel_node, const KernelBuildInfoPtr &select_kernel_buil… in IsSupportedByAICPU()
144 [&select_kernel_build_info](const kernel::KernelBuildInfoPtr item) { in IsSupportedByAICPU()
150 bool IsSupportedByAICore(const AnfNodePtr &kernel_node, const KernelBuildInfoPtr &select_kernel_bui… in IsSupportedByAICore()
158 [&select_kernel_build_info](const kernel::KernelBuildInfoPtr item) { in IsSupportedByAICore()
|
| D | kernel_query.h | 31 bool IsSupportedByAICPU(const AnfNodePtr &kernel_node, const KernelBuildInfoPtr &select_kernel_buil…
32 bool IsSupportedByAICore(const AnfNodePtr &kernel_node, const KernelBuildInfoPtr &select_kernel_bui…
|
| D | kernel_build_info.h | 122 using KernelBuildInfoPtr = std::shared_ptr<KernelBuildInfo>; variable
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ |
| D | ascend_helper.h | 45 const kernel::KernelBuildInfoPtr &select_kernel_build_info) { in CheckAICoreSupported()
49 const kernel::KernelBuildInfoPtr &select_kernel_build_info) { in CheckAICPUSupported()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/pass/ |
| D | replace_node_by_proxy.h | 37 kernel::KernelBuildInfoPtr GenerateKernelBuildInfo(const CNodePtr &cnode);
|
| D | replace_node_by_proxy.cc | 25 kernel::KernelBuildInfoPtr ReplaceNodeByProxy::GenerateKernelBuildInfo(const CNodePtr &cnode) { in GenerateKernelBuildInfo()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fusion/ |
| D | layer_norm_beta_gamma_backprop_fusion.cc | 55 bool CheckKernelBuildInfo(const CNodePtr &cnode, const kernel::KernelBuildInfoPtr &kernel_info) { in CheckKernelBuildInfo()
136 [&cnode](const kernel::KernelBuildInfoPtr &candidate_kernel_build_info) { in Process()
|
| D | parameter_and_transop_fusion.cc | 61 kernel::KernelBuildInfoPtr GetKernelBuildInfo(const CNodePtr &cast, const string &format, TypeId in… in GetKernelBuildInfo()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/ |
| D | graph_kernel_helper.h | 66 kernel::KernelBuildInfoPtr BuildSelectKernelBuildInfo(const std::vector<std::string> &inputs_format,
70 kernel::KernelBuildInfoPtr BuildSelectKernelBuildInfo(const std::vector<std::string> &inputs_format,
|
| /third_party/mindspore/mindspore/ccsrc/ps/ |
| D | util.h | 62 …static kernel::KernelBuildInfoPtr GenerateKernelBuildInfo(const std::vector<AnfNodePtr> &node_list…
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/buffer_fusion/ |
| D | fusion_base_pass.h | 55 kernel::KernelBuildInfoPtr kernel_build_info;
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/ |
| D | add_relu_v2_fusion.cc | 30 kernel::KernelBuildInfoPtr GenerateKernelBuildInfo(CNodePtr node) { in GenerateKernelBuildInfo()
|
| D | matmul_biasadd_fusion.cc | 30 kernel::KernelBuildInfoPtr GenerateKernelBuildInfo(CNodePtr node) { in GenerateKernelBuildInfo()
|
| D | add_relu_grad_v2_fusion.cc | 30 kernel::KernelBuildInfoPtr GenerateKernelBuildInfo(CNodePtr node) { in GenerateKernelBuildInfo()
|
| D | relu_v2_pass.cc | 42 kernel::KernelBuildInfoPtr GenerateKernelBuildInfo(CNodePtr node) { in GenerateKernelBuildInfo()
|
| D | adam_fusion.cc | 30 kernel::KernelBuildInfoPtr GenerateKernelBuildInfo(CNodePtr node) { in GenerateKernelBuildInfo()
|
| /third_party/mindspore/tests/ut/cpp/pre_activate/ascend/ir_fusion/ |
| D | transpose_transdata_fusion_test.cc | 45 …bool CheckAICoreSupported(const AnfNodePtr &anf_node, const kernel::KernelBuildInfoPtr &select_ker… in CheckAICoreSupported()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/ |
| D | gpu_kernel_factory.h | 43 bool SearchRegistered(const std::string &kernel_name, const KernelBuildInfoPtr &kernel_info);
|
| D | gpu_kernel_factory.cc | 192 bool GpuKernelFactory::SearchRegistered(const std::string &kernel_name, const KernelBuildInfoPtr &k… in SearchRegistered()
|
| /third_party/mindspore/tests/ut/cpp/pre_activate/ascend/ir_fission/ |
| D | topk_split_test.cc | 58 …bool CheckAICoreSupported(const AnfNodePtr &anf_node, const kernel::KernelBuildInfoPtr &select_ker… in CheckAICoreSupported()
|
| /third_party/mindspore/mindspore/ccsrc/backend/session/ |
| D | anf_runtime_algorithm.h | 228 …static void SetSelectKernelBuildInfo(const kernel::KernelBuildInfoPtr &select_kernel_build_info, A…
230 static kernel::KernelBuildInfoPtr GetSelectKernelBuildInfo(const AnfNodePtr &node);
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/ |
| D | topk_split.cc | 82 kernel::KernelBuildInfoPtr CreateKernelBuildInfo() { in CreateKernelBuildInfo()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/common/ |
| D | helper.h | 225 kernel::KernelBuildInfoPtr GenerateKernelBuildInfo(const std::vector<AnfNodePtr> &node_list);
|
12