Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/
Dcudnn_inplace_fusion.cc125 auto first_node_prim = AnfAlgo::GetCNodePrimitive(first_node); in GetCoverIndex() local
126 MS_EXCEPTION_IF_NULL(first_node_prim); in GetCoverIndex()
127 auto first_node_channel = first_node_prim.get()->GetAttr("out_channel"); in GetCoverIndex()