Searched refs:kernel_info_ptr (Results 1 – 9 of 9) sorted by relevance
87 eqv[node]->set_kernel_info(node->kernel_info_ptr()); in RefSubGraphNode()139 eqv[n]->set_kernel_info(n->kernel_info_ptr()); in TransformSegmentToAnfGraph()
65 new_cnode->set_kernel_info(cnode->kernel_info_ptr()); in ProcessNode()
47 node->set_kernel_info(cnode->kernel_info_ptr()); in CloneCNode()
62 parameter->set_kernel_info(new_input->kernel_info_ptr()); in ProcessOriginCNode()
356 parameter->set_kernel_info(new_input->kernel_info_ptr()); in ProcessOriginCNode()
94 cp_node->set_kernel_info(cnode->kernel_info_ptr()); in NewRecomputeNode()
239 parameter->set_kernel_info(vnode->kernel_info_ptr()); in ConvertNonscalarTensorToParameter()
212 auto kernel_info_ptr = kernel_info_list[info_index]; in ChooseMatchedKernelInfo() local213 MS_EXCEPTION_IF_NULL(kernel_info_ptr); in ChooseMatchedKernelInfo()214 UpdateCurMatchCounts(*kernel_info_ptr, kernel_node, &cur_kernel_info_match_counts); in ChooseMatchedKernelInfo()
129 const KernelInfoDevicePtr &kernel_info_ptr() { return kernel_info_; } in kernel_info_ptr() function348 set_kernel_info(node->kernel_info_ptr()); in CloneCNodeInfo()