| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/ |
| D | kernel_registry.cc | 225 auto lite_kernel = creator(in_tensors, out_tensors, parameter, ctx, *key); in GetLiteKernel() local
236 auto *lite_kernel = nnacl::NNACLKernelRegistry(parameter, in_tensors, out_tensors, ctx, *key); in GetLiteKernel() local
263 auto lite_kernel = GetLiteKernel(in_tensors, out_tensors, ctx, &modify_key, parameter); in GetKernelExec() local
|
| D | runtime_packed_node_pass.cc | 288 auto lite_kernel = static_cast<kernel::LiteKernel *>(kernel); in PackedMatmulKernelExec() local
|
| D | kernel_exec_util.cc | 383 …auto lite_kernel = new (std::nothrow) LiteKernel(nullptr, input_tensors, output_tensors, &context); in CreateSubGraphKernel() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/test/ut/src/runtime/kernel/opencl/ |
| D | fill_tests.cc | 60 auto lite_kernel = in TEST_F() local
130 auto lite_kernel = in TEST_F() local
|
| D | cast_tests.cc | 73 auto lite_kernel = std::make_shared<kernel::CastOpenCLKernel>(param, inputs, outputs, nullptr); in TEST_F() local
182 auto lite_kernel = std::make_shared<kernel::CastOpenCLKernel>(param, inputs, outputs, nullptr); in TEST_F() local
|
| D | common.cc | 108 auto *lite_kernel = creator(kernel_inputs, outputs, op_parameter, nullptr, key); in TestMain() local
268 auto *lite_kernel = creator(kernel_inputs, {&output}, op_parameter, nullptr, key); in TestMain() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/kernel/nnacl/ |
| D | nnacl_base_kernel.h | 30 explicit NNACLBaseKernel(std::shared_ptr<LiteKernel> lite_kernel) in NNACLBaseKernel()
|
| D | nnacl_lib.cc | 89 …auto lite_kernel = lite::KernelRegistry::GetInstance()->GetLiteKernel(inputs, outputs, ctx, &key, … in CreateKernel() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/control_flow/kernel/ |
| D | identity_kernel.cc | 99 auto lite_kernel = new IdentityKernel(param, in_tensors, out_tensors, ctx); in Create() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/pass/format_pass/ |
| D | format_pass.cc | 47 …auto lite_kernel = KernelRegistry::GetInstance()->GetLiteKernel({input}, {output}, ctx, &format_tr… in DefaultCreateFormatTransFunc() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/ |
| D | converter_packed_node.cc | 75 auto *lite_kernel = PackDataWrapper::GetInstance().GetPackedKernel(cnode->name); in ReplaceMatMulFusionToCustom() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/graph_compiler/ |
| D | single_graph_scheduler.cc | 173 …auto lite_kernel = KernelRegistry::GetInstance()->GetLiteKernel({input}, {output}, ctx, &format_tr… in CreateFormatTransFunc() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/const_fold/ |
| D | fold_utils.cc | 91 auto lite_kernel = item.second(in_tensors, out_tensors, parameter, ctx, key); in GetHighAccuracyKernel() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/opencl/ |
| D | opencl_fusion.cc | 544 …auto lite_kernel = std::make_shared<FusionEltwiseOpenCLKernel>(reinterpret_cast<OpParameter *>(par… in CreateEltwiseKernelReplaceOld() local
|