Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/
Dcudnn_inplace_fusion.cc181 const auto &acc_input = acc_node->input(1)->cast<CNodePtr>(); in CheckInplaceNodeInputs() local
182 if (acc_input == nullptr) { in CheckInplaceNodeInputs()
185 bool ret = ExistRoute(acc_input, cover_node); in CheckInplaceNodeInputs()
187 auto new_input = graph->NewCNode(acc_input->inputs()); in CheckInplaceNodeInputs()
189 new_input->set_abstract(acc_input->abstract()); in CheckInplaceNodeInputs()
190 CopyKernelInfo(acc_input, new_input); in CheckInplaceNodeInputs()