Home
last modified time | relevance | path

Searched defs:kernel_exec (Results 1 – 12 of 12) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/graph_compiler/
Dsingle_graph_scheduler.cc86 auto kernel_exec = new kernel::ExtendRTKernelExec(delegate_kernel); in CreateDelegateKernel() local
115 auto kernel_exec = in SelectKernel() local
185 …auto *kernel_exec = new (std::nothrow) kernel::ExtendRTKernelExec(std::shared_ptr<kernel::MSKernel… in CreateFormatTransFunc() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/const_fold/
Dfold_utils.cc98 auto *kernel_exec = new (std::nothrow) kernel::KernelExec(shared_kernel); in GetHighAccuracyKernel() local
134 kernel::KernelExec *kernel_exec = nullptr; in GetKernelExec() local
296 …auto kernel_exec = GetKernelExec(input_tensors, &output_tensors, cnode, context_.get(), ms_context… in DoConstantFold() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/
Dkernel_registry.cc187 auto *kernel_exec = new (std::nothrow) kernel::KernelExec(base_kernel); in GetCustomKernel() local
266 auto *kernel_exec = new (std::nothrow) kernel::KernelExec(shared_kernel); in GetKernelExec() local
Druntime_packed_node_pass.cc261 int PackedMatmulKernelExec(kernel::KernelExec *kernel_exec, const std::vector<Tensor *> &tensors) { in PackedMatmulKernelExec()
333 int PackKernelExec(kernel::KernelExec *kernel_exec, const std::vector<Tensor *> &tensors) { in PackKernelExec()
Dcache_session.cc324 auto *kernel_exec = new (std::nothrow) kernel::KernelExec(shared_kernel); in CreateFullModelKernel() local
Dscheduler.cc555 auto kernel_exec = new (std::nothrow) kernel::KernelExec(shared_kernel); in ReplaceDelegateKernels() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/mindir_loader/mindir_model/
Dmindir_model.cc88 kernel::KernelExec *kernel_exec = new kernel::KernelExec(inner_kernel); in FindBackendKernel() local
99 mindspore::kernel::KernelExec *kernel_exec = nullptr; in FindLiteKernel() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/control_flow/kernel/
Didentity_kernel.cc103 auto *kernel_exec = new KernelExec(shared_kernel); in Create() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/kernel/default/
Ddefault_kernel_lib.cc65 auto *kernel_exec = KernelLib::CreateKernelExec(spec, inputs, outputs, ctx); in CreateKernelExec() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/kernel/nnacl/
Dnnacl_lib.cc100 auto *kernel_exec = KernelLib::CreateKernelExec(spec, inputs, outputs, ctx); in CreateKernelExec() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/pass/format_pass/
Dformat_pass.cc55 …auto *kernel_exec = new (std::nothrow) kernel::KernelExec(std::shared_ptr<kernel::MSKernel>(kernel… in DefaultCreateFormatTransFunc() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/kernel/
Dkernel_lib.h64 …auto *kernel_exec = new (std::nothrow) ExtendRTKernelExec(std::shared_ptr<BaseKernel>(base_kernel)… in CreateKernelExec() local