Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/
Dgpu_launch_mul.cc35 void GPULaunchMul::KernelBuild(const std::shared_ptr<session::KernelGraph> &kernel_graph) { in KernelBuild() function in mindspore::device::gpu::GPULaunchMul
Dgpu_launch_kernel.cc55 void GPULaunchkernel::KernelBuild(const std::shared_ptr<session::KernelGraph> &kernel_graph) { in KernelBuild() function in mindspore::device::gpu::GPULaunchkernel
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/
Dascend_launch_kernel.cc45 void AscendLaunchKernel::KernelBuild(const std::shared_ptr<session::KernelGraph> &kernel_graph) { in KernelBuild() function in mindspore::device::ascend::AscendLaunchKernel
Dascend_launch_mul.cc36 void AscendLaunchMul::KernelBuild(const std::shared_ptr<session::KernelGraph> &kernel_graph) { in KernelBuild() function in mindspore::device::ascend::AscendLaunchMul
Dascend_launch_transdata.cc37 void AscendLaunchTransData::KernelBuild(const std::shared_ptr<session::KernelGraph> &kernel_graph) { in KernelBuild() function in mindspore::device::ascend::AscendLaunchTransData
Dascend_launch_atomic_clean.cc36 void AscendLaunchAtomicClean::KernelBuild(const std::shared_ptr<session::KernelGraph> &kernel_graph… in KernelBuild() function in mindspore::device::ascend::AscendLaunchAtomicClean
Dkernel_build_ascend.cc286 bool KernelBuild(const std::vector<CNodePtr> &kernels) { in KernelBuild() function