| /third_party/mindspore/mindspore/lite/test/ut/src/runtime/ |
| D | runtime_pass_tests.cc | 51 reg->GetKernel(conv_in, conv_out, ctx, nullptr, conv_desc, op_conv, &conv_kernel, nullptr); in Nc4hw4PassConstruct()
64 …reg->GetKernel(trans_in, trans_out, ctx, nullptr, trans_desc, trans_param, &trans_kernel, nullptr); in Nc4hw4PassConstruct()
77 reg->GetKernel(in_in, in_out, ctx, nullptr, in_desc, in_param, &in_kernel, nullptr); in Nc4hw4PassConstruct()
90 …reg->GetKernel(trans2_in, trans2_out, ctx, nullptr, trans2_desc, trans2_param, &trans2_kernel, nul… in Nc4hw4PassConstruct()
105 reg->GetKernel(conv2_in, conv2_out, ctx, nullptr, conv2_desc, op2_conv, &conv2_kernel, nullptr); in Nc4hw4PassConstruct()
137 reg->GetKernel(conv1_in, conv1_out, ctx, nullptr, conv1_desc, op1_conv, &conv1_kernel, nullptr); in ConvNormC4PassConstruct()
150 reg->GetKernel(in_in, in_out, ctx, nullptr, in_desc, in_param, &in_kernel, nullptr); in ConvNormC4PassConstruct()
166 reg->GetKernel(conv2_in, conv2_out, ctx, nullptr, conv2_desc, op2_conv, &conv2_kernel, nullptr); in ConvNormC4PassConstruct()
177 reg->GetKernel(act_in, act_out, ctx, nullptr, act_desc, act_param, &act_kernel, nullptr); in ConvNormC4PassConstruct()
190 reg->GetKernel(in2_in, in2_out, ctx, nullptr, in2_desc, in2_param, &in2_kernel, nullptr); in ConvNormC4PassConstruct()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/akg/gpu/ |
| D | akg_gpu_kernel_mod.cc | 39 kernel_pack->GetKernel() == nullptr || kernel_pack->GetKernel()->contents == nullptr) { in GetFunction()
74 …CUresult result = cuModuleLoadDataEx(&module, kernel_pack->GetKernel()->contents, 1, options, valu… in GetFunction()
|
| /third_party/mindspore/mindspore/lite/src/ |
| D | kernel_registry.h | 45 int GetKernel(const std::vector<Tensor *> &in_tensors, const std::vector<Tensor *> &out_tensors,
|
| D | kernel_registry.cc | 177 int KernelRegistry::GetKernel(const std::vector<Tensor *> &in_tensors, const std::vector<Tensor *> … in GetKernel() function in mindspore::lite::KernelRegistry
|
| D | scheduler.cc | 900 …ret = KernelRegistry::GetInstance()->GetKernel(in_tensors, out_tensors, context_, ms_context_, cpu… in FindCpuKernel()
946 …ret = KernelRegistry::GetInstance()->GetKernel(in_tensors, out_tensors, context_, ms_context_, gpu… in FindGpuKernel()
968 …ret = KernelRegistry::GetInstance()->GetKernel(in_tensors, out_tensors, context_, ms_context_, des… in FindProviderKernel()
977 …ret = KernelRegistry::GetInstance()->GetKernel(in_tensors, out_tensors, context_, ms_context_, des… in FindProviderKernel()
994 …ret = KernelRegistry::GetInstance()->GetKernel(in_tensors, out_tensors, context_, ms_context_, des… in FindProviderKernel()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/ |
| D | kernel.h | 138 const FlexArray *GetKernel() const { return kernel_; } in GetKernel() function
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/ |
| D | tbe_utils.cc | 281 auto kernel = kernel_pack.GetKernel(); in GenFuncStub()
307 if (BinaryRegister((*kernel_pack.GetKernel()), &module, magic, dynamic_flag) != 0) { in GenFuncStub()
|
| /third_party/mindspore/mindspore/lite/tools/optimizer/fusion/ |
| D | constant_folding_fusion.cc | 203 …ret = lite::KernelRegistry::GetInstance()->GetKernel(inputs, *outputs, context, ms_context, desc, … in GetLiteKernel()
|