Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/
Dcudnn_inplace_fusion.cc107 const CNodePtr &cover_node = inplace_node[cover_index].node->cast<CNodePtr>(); in ExistDependencyFromAcc2Cover() local
109 MS_EXCEPTION_IF_NULL(cover_node); in ExistDependencyFromAcc2Cover()
111 return ExistRoute(acc_node, cover_node); in ExistDependencyFromAcc2Cover()
177 const CNodePtr &cover_node = inplace_node->at(cover_index).node->cast<CNodePtr>(); in CheckInplaceNodeInputs() local
179 MS_EXCEPTION_IF_NULL(cover_node); in CheckInplaceNodeInputs()
185 bool ret = ExistRoute(acc_input, cover_node); in CheckInplaceNodeInputs()
226 auto cover_node = (*inplace_node)[cover_index].node; in SetNodeAttr() local
231 acc_node_input, cover_node}; in SetNodeAttr()