Searched refs:set_kernel_mod (Results 1 – 7 of 7) sorted by relevance
170 bool set_kernel_mod) { in TaskFinishProcess() argument179 if (set_kernel_mod) { in TaskFinishProcess()191 if (set_kernel_mod) { in TaskFinishProcess()
61 bool set_kernel_mod = true);
60 void set_kernel_mod(const kernel::KernelModPtr &kernel_mod);
111 void KernelInfo::set_kernel_mod(const kernel::KernelModPtr &kernel_mod) { kernel_mod_ = kernel_mod;… in set_kernel_mod() function in mindspore::device::KernelInfo
91 kernel_info_first->set_kernel_mod(kernel_mod_first); in CreateKernelGraph()121 kernel_info_floor->set_kernel_mod(kernel_mod_floor); in CreateKernelGraph()
706 d_kernel_info->set_kernel_mod(nullptr); in TEST_F()
1399 kernel_info->set_kernel_mod(kernel_mod); in SetKernelMod()