Searched refs:base_kernel (Results 1 – 3 of 3) sorted by relevance
| /third_party/mindspore/mindspore/lite/src/ |
| D | kernel_registry.cc | 155 …auto base_kernel = creator(LiteTensorsToMSTensors(in_tensors), LiteTensorsToMSTensors(out_tensors), in GetCustomKernel() local
157 if (base_kernel != nullptr) { in GetCustomKernel()
158 auto *lite_kernel = new (std::nothrow) kernel::LiteKernel(base_kernel); in GetCustomKernel()
|
| /third_party/mindspore/mindspore/_extends/parallel_compile/tbe_compiler/ |
| D | tuner.py | 298 base_kernel = './kernel_meta/' + kernel_name + '.o'
319 … ret = dispatch_single_tune_task(graph_id, task_id, l1size, base_kernel, kernel_name, full_name,
341 base_kernel = './kernel_meta/' + kernel_name + '.o'
358 … ret = dispatch_fusion_tune_task(graph_id, task_id, l1size, base_kernel, kernel_name, full_name,
|
| D | tbe_adapter.py | 536 base_kernel = job.content["SocInfo"]["op_debug_dir"] + "/kernel_meta/" + op_kernel_name + ".o"
539 …res = dispatch_single_tune_task(job.source_id, job.id, l1_size, base_kernel, op_kernel_name, full_…
568 base_kernel = job.content["SocInfo"]["op_debug_dir"] + "/kernel_meta/" + op_kernel_name + ".o"
578 …res = dispatch_fusion_tune_task(job.source_id, job.id, l1_size, base_kernel, op_kernel_name, op_mo…
|