Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/
Dcudnn_inplace_fusion.cc191 auto new_inplace_node = graph->NewCNode({acc_node->input(0), new_input, acc_node->input(2)}); in CheckInplaceNodeInputs() local
192 MS_EXCEPTION_IF_NULL(new_inplace_node); in CheckInplaceNodeInputs()
193 new_inplace_node->set_abstract(acc_node->abstract()); in CheckInplaceNodeInputs()
194 CopyKernelInfo(acc_node, new_inplace_node); in CheckInplaceNodeInputs()
197 manager->Replace(acc_node, new_inplace_node); in CheckInplaceNodeInputs()
198 (*inplace_node)[acc_index].node = new_inplace_node; in CheckInplaceNodeInputs()