Home
last modified time | relevance | path

Searched defs:kernel_mod_ptr (Results 1 – 6 of 6) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/akg/ascend/
Dakg_ascend_kernel_build.cc39 auto kernel_mod_ptr = std::make_shared<AkgKernelMod>(kernel_pack); in AkgSetKernelMod() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/aicpu/
Daicpu_kernel_build.cc77 …IOSize(const std::shared_ptr<AnfNode> &anf_node, const std::shared_ptr<AicpuOpKernelMod> &kernel_m… in SetIOSize()
267 const std::shared_ptr<AicpuOpKernelMod> &kernel_mod_ptr) { in CreateNodeDefBytes()
362 …xtInfo(const std::shared_ptr<AnfNode> &anf_node, const std::shared_ptr<AicpuOpKernelMod> &kernel_m… in CreateExtInfo()
412 auto kernel_mod_ptr = std::make_shared<AicpuOpKernelMod>(); in AicpuOpBuild() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/akg/gpu/
Dakg_gpu_kernel_build.cc41 auto kernel_mod_ptr = std::make_shared<GpuKernelMod>(kernel_pack); in AkgSetKernelMod() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/
Dtbe_kernel_parallel_build.cc277 auto kernel_mod_ptr = GenKernelMod(input_size_list, output_size_list, cached_kernel_pack); in SearchInCache() local
291 auto kernel_mod_ptr = std::make_shared<TbeKernelMod>(kernel_pack); in GenKernelMod() local
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/
Dkernel_build_ascend.cc47 kernel::KernelModPtr kernel_mod_ptr = nullptr; in SerialCompileImpl() local
124 kernel::KernelModPtr kernel_mod_ptr = SerialCompileImpl(anf_node); in KernelBuildParallelCompile() local
/third_party/mindspore/mindspore/ccsrc/ps/ps_cache/ascend/
Dascend_ps_cache.cc92 const std::shared_ptr<kernel::AicpuOpKernelMod> &kernel_mod_ptr) { in SetNodedefProto()