Searched refs:SetPrimAttr (Results 1 – 1 of 1) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/ |
| D | cudnn_inplace_fusion.cc | 58 void SetPrimAttr(AnfNodePtr inplace_node, const string &key, const T &value) { in SetPrimAttr() function
208 SetPrimAttr(aggregate_node.node, "aggregate", true); in SetNodeAttr()
209 SetPrimAttr(aggregate_node.node, "aggregate_input_index", aggregate_node.index); in SetNodeAttr()
210 SetPrimAttr(skip_node, "skip", true); in SetNodeAttr()
220 SetPrimAttr(node, "inplace_algo", algo); in SetNodeAttr()
221 SetPrimAttr(node, "inplace_group", group); in SetNodeAttr()
222 SetPrimAttr(node, "inplace_output_index", (*inplace_node)[i].index); in SetNodeAttr()
|