Searched refs:GenKernelMod (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/ |
| D | tbe_kernel_parallel_build.cc | 189 …auto kernel_mod = GenKernelMod(task_iter->second.input_size_list, task_iter->second.output_size_li… in TaskFinishProcess()
261 …auto kernel_mode = GenKernelMod(task_info.input_size_list, task_info.output_size_list, kernel_pack… in GenSameFusionOpKernelMod()
277 auto kernel_mod_ptr = GenKernelMod(input_size_list, output_size_list, cached_kernel_pack); in SearchInCache()
286 KernelModPtr ParallelBuildManager::GenKernelMod(const std::vector<size_t> &input_size_list, in GenKernelMod() function in mindspore::kernel::ParallelBuildManager
|
| D | tbe_kernel_parallel_build.h | 62 …KernelModPtr GenKernelMod(const std::vector<size_t> &input_size_list, const std::vector<size_t> &o…
|
| D | ascend_kernel_compile.cc | 681 … auto kernel_mod = build_manager_->GenKernelMod(input_size_list, output_size_list, kernel_pack); in AscendFusionOpCompile()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/ |
| D | kernel_fusion.cc | 97 auto kernel_mod = build_manger->GenKernelMod(input_size_list, output_size_list, kernel_pack); in KernelFusion()
|