Home
last modified time | relevance | path

Searched defs:SetKernelMod (Results 1 – 7 of 7) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/akg/
Dakg_cpu_kernel_build.cc24 void AkgCpuKernelBuilder::SetKernelMod(const KernelPackPtr &kernel_pack, const GraphKernelJsonGener… in SetKernelMod() function in mindspore::kernel::AkgCpuKernelBuilder
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/akg/
Dakg_gpu_kernel_build.cc30 void AkgGpuKernelBuilder::SetKernelMod(const KernelPackPtr &kernel_pack, const GraphKernelJsonGener… in SetKernelMod() function in mindspore::kernel::AkgGpuKernelBuilder
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/akg/
Dakg_ascend_kernel_build.cc41 void AkgAscendKernelBuilder::SetKernelMod(const KernelPackPtr &kernel_pack, in SetKernelMod() function in mindspore::kernel::AkgAscendKernelBuilder
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/dynamic_akg/
Ddynamic_akg_gpu_kernel_build.cc30 void DynamicAkgGpuKernelBuilder::SetKernelMod(const KernelPackPtr &kernel_pack, in SetKernelMod() function in mindspore::kernel::DynamicAkgGpuKernelBuilder
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/dynamic_akg/
Ddynamic_akg_cpu_kernel_build.cc26 void DynamicAkgCpuKernelBuilder::SetKernelMod(const KernelPackPtr &kernel_pack, in SetKernelMod() function in mindspore::kernel::DynamicAkgCpuKernelBuilder
/third_party/mindspore/mindspore-src/source/tests/ut/cpp/session/
Danf_runtime_algorithm_test.cc726 TEST_F(AnfRuntimeAlgorithmTest, SetKernelMod) { in TEST_F() argument
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/session/
Danf_runtime_algorithm.cc1288 void AnfRuntimeAlgorithm::SetKernelMod(const KernelModPtr &kernel_mod, AnfNode *node) { in SetKernelMod() function in mindspore::session::AnfRuntimeAlgorithm