Home
last modified time | relevance | path

Searched refs:clear_zero (Results 1 – 1 of 1) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/
Dkernel_build_ascend.cc170 CNodePtr clear_zero = kernel_graph->NewCNode(inputs); in AddTbeClearZeroNode() local
171 MS_EXCEPTION_IF_NULL(clear_zero); in AddTbeClearZeroNode()
174 clear_zero->set_abstract(abstract); in AddTbeClearZeroNode()
178 AnfAlgo::SetSelectKernelBuildInfo(builder->Build(), clear_zero.get()); in AddTbeClearZeroNode()
180 AnfAlgo::SetNodeAttr(kAttrAtomicAddMemSize, MakeValue(clean_size), clear_zero); in AddTbeClearZeroNode()
181 …AnfAlgo::SetStreamDistinctionLabel(AnfAlgo::GetStreamDistinctionLabel(pre_node.get()), clear_zero.… in AddTbeClearZeroNode()
182 new_nodes->push_back(clear_zero); in AddTbeClearZeroNode()
197 CNodePtr clear_zero = kernel_graph->NewCNode(inputs); in AddFusionTbeClearZeroNode() local
198 MS_EXCEPTION_IF_NULL(clear_zero); in AddFusionTbeClearZeroNode()
201 clear_zero->set_abstract(abstract); in AddFusionTbeClearZeroNode()
[all …]