Searched defs:kernel_pack (Results 1 – 12 of 12) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/akg/ |
| D | akg_gpu_kernel_mod.cc | 28 CUresult AkgGpuKernelManager::GetFunction(const KernelPackPtr &kernel_pack, bool force_reload, in GetFunction()
84 AkgGpuKernelMod::AkgGpuKernelMod(const KernelPackPtr &kernel_pack) : kernel_pack_(kernel_pack) { in AkgGpuKernelMod()
|
| D | akg_gpu_kernel_build.cc | 30 void AkgGpuKernelBuilder::SetKernelMod(const KernelPackPtr &kernel_pack, const GraphKernelJsonGener… in SetKernelMod()
|
| D | akg_gpu_kernel_mod.h | 93 explicit AkgGpuKernelModDebug(const KernelPackPtr &kernel_pack) : AkgGpuKernelMod(kernel_pack) {} in AkgGpuKernelModDebug()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/akg/ |
| D | akg_cpu_kernel_build.cc | 24 void AkgCpuKernelBuilder::SetKernelMod(const KernelPackPtr &kernel_pack, const GraphKernelJsonGener… in SetKernelMod()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/akg/ |
| D | akg_ascend_kernel_build.cc | 41 void AkgAscendKernelBuilder::SetKernelMod(const KernelPackPtr &kernel_pack, in SetKernelMod()
|
| D | akg_ascend_kernel_mod.cc | 38 AkgKernelMod::AkgKernelMod(const KernelPackPtr &kernel_pack, const AnfNodePtr &anf_node_ptr) in AkgKernelMod()
|
| D | akg_utils.cc | 57 void *KernelManager::GenFuncStub(const mindspore::kernel::KernelPack &kernel_pack, bool force_reloa… in GenFuncStub()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/dynamic_akg/ |
| D | dynamic_akg_gpu_kernel_build.cc | 30 void DynamicAkgGpuKernelBuilder::SetKernelMod(const KernelPackPtr &kernel_pack, in SetKernelMod()
|
| D | dynamic_akg_gpu_kernel_mod.cc | 72 CUresult DynamicAkgGpuKernelManager::GetFunction(const KernelPackPtr &kernel_pack, bool force_reloa… in GetFunction()
111 DynamicAkgGpuKernelMod::DynamicAkgGpuKernelMod(const KernelPackPtr &kernel_pack) : kernel_pack_(ker… in DynamicAkgGpuKernelMod()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/dynamic_akg/ |
| D | dynamic_akg_cpu_kernel_build.cc | 26 void DynamicAkgCpuKernelBuilder::SetKernelMod(const KernelPackPtr &kernel_pack, in SetKernelMod()
|
| D | dynamic_akg_cpu_kernel_mod.cc | 121 DynamicAkgCpuKernelMod::DynamicAkgCpuKernelMod(const KernelPackPtr &kernel_pack) { in DynamicAkgCpuKernelMod()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/kernel/ |
| D | framework_utils.cc | 254 KernelPackPtr kernel_pack = std::make_shared<KernelPack>(); in SearchCache() local
277 KernelPackPtr kernel_pack = std::make_shared<KernelPack>(); in InsertCache() local
|