Home
last modified time | relevance | path

Searched defs:apply_kernel (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/hal/device/
Dgpu_kernel_build.cc37 void SetGpuRefMapToKernelInfo(const CNodePtr &apply_kernel, const std::vector<kernel::KernelAttr> &… in SetGpuRefMapToKernelInfo()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/
Dgpu_kernel_factory.cc228 … *NativeGpuKernelModFactory::Create(const std::string &kernel_name, const CNodePtr &apply_kernel) { in Create()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/kernel/
Dcommon_utils.cc754 void SetCpuRefMapToKernelInfo(const CNodePtr &apply_kernel, const std::vector<KernelAttr> &apply_ke… in SetCpuRefMapToKernelInfo()