Searched refs:aggregate_node (Results 1 – 1 of 1) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/ |
| D | cudnn_inplace_fusion.cc | 202 void SetNodeAttr(AnfNodeIndex aggregate_node, AnfNodePtr skip_node, std::vector<AnfNodeIndex> *inpl… in SetNodeAttr() argument
208 SetPrimAttr(aggregate_node.node, "aggregate", true); in SetNodeAttr()
209 SetPrimAttr(aggregate_node.node, "aggregate_input_index", aggregate_node.index); in SetNodeAttr()
311 AnfNodeIndex aggregate_node; in Run() local
315 if (!PatternMatch(graph, node, &aggregate_node, &skip_node, &inplace_node)) { in Run()
333 MS_LOG(INFO) << "[inplace optimizer] aggregate node: " << aggregate_node.index << ", " in Run()
334 … << aggregate_node.node->DebugString() << "; skip node: " << skip_node->DebugString() << std::endl in Run()
340 SetNodeAttr(aggregate_node, skip_node, &inplace_node, graph); in Run()
|