Searched refs:kernel_mode (Results 1 – 2 of 2) sorted by relevance
261 …auto kernel_mode = GenKernelMod(task_info.input_size_list, task_info.output_size_list, kernel_pack… in GenSameFusionOpKernelMod() local262 if (kernel_mode != nullptr) { in GenSameFusionOpKernelMod()263 (*kernel_mode_ret)[task_info.scope_id] = kernel_mode; in GenSameFusionOpKernelMod()
373 auto kernel_mode = AnfAlgo::GetKernelMod(cnode); in FetchMemUnitSize() local374 MS_EXCEPTION_IF_NULL(kernel_mode); in FetchMemUnitSize()380 const auto &input_size_list = kernel_mode->GetInputSizeList(); in FetchMemUnitSize()381 const auto &output_size_list = kernel_mode->GetOutputSizeList(); in FetchMemUnitSize()382 const auto &workspace_size_list = kernel_mode->GetWorkspaceSizeList(); in FetchMemUnitSize()