Searched refs:clean_node (Results 1 – 5 of 5) sorted by relevance
100 auto clean_node = atomic_clean_graph_->execution_order()[0]; in ConstructKernelGraphAndSetAttr() local103 MS_EXCEPTION_IF_NULL(clean_node); in ConstructKernelGraphAndSetAttr()104 clean_node->set_abstract(abstract); in ConstructKernelGraphAndSetAttr()109 AnfAlgo::SetSelectKernelBuildInfo(builder->Build(), clean_node.get()); in ConstructKernelGraphAndSetAttr()112 AnfAlgo::SetNodeAttr(kAttrAtomicAddMemSize, MakeValue(clean_size), clean_node); in ConstructKernelGraphAndSetAttr()
30 static inline void clean_node(unsigned long *array) in clean_node() function
82 …void AddDepend(const FuncGraphPtr &main_graph, const AnfNodePtr &clean_node, const AnfNodePtr &com…
369 void AtomicCleanInsertter::AddDepend(const FuncGraphPtr &main_graph, const AnfNodePtr &clean_node, in AddDepend() argument372 AnfNodePtrList d_inputs = {NewValueNode(prim::kPrimDepend), clean_node, composite_node}; in AddDepend()374 depend_cnode->set_abstract(clean_node->abstract()); in AddDepend()
556 clean_node(nmask); in test_ksm_merge_across_nodes()