Home
last modified time | relevance | path

Searched defs:BuildKernel (Results 1 – 8 of 8) sorted by relevance

/third_party/mindspore/mindspore/lite/src/runtime/kernel/opencl/kernel/
Dpooling2d.cc56 int PoolingOpenCLKernel::BuildKernel() { in BuildKernel() function in mindspore::kernel::PoolingOpenCLKernel
Dwinograd.cc81 int WinogradOpenCLKernel::BuildKernel() { in BuildKernel() function in mindspore::kernel::WinogradOpenCLKernel
Dconv2d.cc156 int Conv2DOpenCLKernel::BuildKernel() { in BuildKernel() function in mindspore::kernel::Conv2DOpenCLKernel
/third_party/mindspore/mindspore/lite/src/runtime/gpu/opencl/
Dopencl_runtime_wrapper.cc45 Status OpenCLRuntimeWrapper::BuildKernel(cl::Kernel *kernel, const std::string &program_name, in BuildKernel() function in mindspore::registry::opencl::OpenCLRuntimeWrapper
Dopencl_runtime.cc387 int OpenCLRuntime::BuildKernel(const cl::Kernel &kernel, const std::string &program_name, in BuildKernel() function in mindspore::lite::opencl::OpenCLRuntime
/third_party/mindspore/mindspore/ccsrc/backend/session/
Dcpu_session.cc364 void CPUSession::BuildKernel(const KernelGraph *kernel_graph) { in BuildKernel() function in mindspore::session::CPUSession
Dgpu_session.cc242 void GPUSession::BuildKernel(const std::shared_ptr<KernelGraph> &kernel_graph) const { in BuildKernel() function in mindspore::session::gpu::GPUSession
Dascend_session.cc1303 void AscendSession::BuildKernel(const std::shared_ptr<KernelGraph> &kernel_graph) const { in BuildKernel() function in mindspore::session::AscendSession
1307 void AscendSession::BuildKernel(const std::vector<CNodePtr> &kernels) { in BuildKernel() function in mindspore::session::AscendSession