Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/host/
Dhost_kernel_build.cc32 auto kernel_mod_ptr = HostKernelFactory::Get(prim->name()); in HostOpBuild() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/akg/
Dakg_cpu_kernel_build.cc26 auto kernel_mod_ptr = std::make_shared<AkgCpuKernelMod>(kernel_pack); in SetKernelMod() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/hccl/
Dhccl_kernel_build.cc33 auto kernel_mod_ptr = HcclKernelFactory::Get(prim->name()); in HcclOpBuild() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/akg/
Dakg_gpu_kernel_build.cc33 …auto kernel_mod_ptr = flags.enable_debug_mode ? std::make_shared<AkgGpuKernelModDebug>(kernel_pack) in SetKernelMod() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/acl/
Dacl_kernel_build.cc34 auto kernel_mod_ptr = std::make_shared<AclKernelMod>(); in AclOpBuild() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/akg/
Dakg_ascend_kernel_build.cc43 auto kernel_mod_ptr = std::make_shared<AkgKernelMod>(kernel_pack, anf_node); in SetKernelMod() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/dynamic_akg/
Ddynamic_akg_gpu_kernel_build.cc33 auto kernel_mod_ptr = std::make_shared<DynamicAkgGpuKernelMod>(kernel_pack); in SetKernelMod() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/dynamic_akg/
Ddynamic_akg_cpu_kernel_build.cc29 auto kernel_mod_ptr = std::make_shared<DynamicAkgCpuKernelMod>(kernel_pack); in SetKernelMod() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/bisheng/
Dbisheng_kernel_build.cc61 auto kernel_mod_ptr = std::make_shared<CustomBiShengKernel>(cnode); in BiShengOpBuild() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/
Daicpu_kernel_build.cc181 …xtInfo(const std::shared_ptr<AnfNode> &anf_node, const std::shared_ptr<AicpuOpKernelMod> &kernel_m… in CreateExtInfo()
250 std::shared_ptr<AicpuOpKernelMod> kernel_mod_ptr = std::make_shared<AicpuOpKernelMod>(); in AicpuOpBuild() local
Daicpu_kernel_load.cc149 const std::shared_ptr<AicpuOpKernelMod> &kernel_mod_ptr) { in LoadAicpuKernelSo()
/third_party/mindspore/mindspore-src/source/tests/ut/cpp/common/
Ddevice_common_test.h158 auto kernel_mod_ptr = std::make_shared<TestKernelMod>(); in CreateKernel() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/hal/hardware/
Dge_kernel_executor.cc145 kernel::KernelModPtr kernel_mod_ptr = nullptr; in GenerateKernelMod() local
963 auto kernel_mod_ptr = kernel::RtOpBuild(node); in CreateEventKernelMod() local