Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore-src/source/mindspore/lite/include/registry/
Dopencl_runtime_wrapper.h131 Status OpenCLRuntimeWrapper::BuildKernel(cl::Kernel *kernel, const std::string &program_name, in BuildKernel() function
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/opencl/kernel/
Dpooling2d.cc56 int PoolingOpenCLKernel::BuildKernel() { in BuildKernel() function in mindspore::kernel::PoolingOpenCLKernel
Dwinograd.cc98 int WinogradOpenCLKernel::BuildKernel() { in BuildKernel() function in mindspore::kernel::WinogradOpenCLKernel
Dconv2d.cc203 int Conv2DOpenCLKernel::BuildKernel() { in BuildKernel() function in mindspore::kernel::Conv2DOpenCLKernel
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/graph_kernel/converter/
Dgraph_kernel_optimization.cc156 GkPassManagerPtr GraphKernelOptimizer::BuildKernel() const { in BuildKernel() function in mindspore::graphkernel::GraphKernelOptimizer
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/gpu/opencl/
Dopencl_runtime_wrapper.cc41 Status OpenCLRuntimeWrapper::BuildKernel(cl::Kernel *kernel, const std::vector<char> &program_name, in BuildKernel() function in mindspore::registry::opencl::OpenCLRuntimeWrapper
Dopencl_runtime.cc423 int OpenCLRuntime::BuildKernel(const cl::Kernel &kernel, const std::string &program_name, in BuildKernel() function in mindspore::lite::opencl::OpenCLRuntime
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/hal/hardware/
Dcpu_session.cc300 void CPUSession::BuildKernel(const KernelGraph *kernel_graph) const { in BuildKernel() function in mindspore::session::CPUSession
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/dvm/
Ddvm_kernel_build.cc613 void BuildKernel(const FuncGraphPtr &graph, const CNodePtr &out_node, in BuildKernel() function in mindspore::kernel::__anond9d285600111::SingleDvmKernelBuilder
681 void BuildKernel(const FuncGraphPtr &graph, const CNodePtr &out_node, in BuildKernel() function in mindspore::kernel::__anond9d285600111::ParallelDvmKernelBuilder
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/hal/hardware/
Dgpu_session.cc235 void GPUSession::BuildKernel(const std::shared_ptr<KernelGraph> &kernel_graph) const { in BuildKernel() function in mindspore::session::gpu::GPUSession