Searched refs:AnfNodeIndex (Results 1 – 1 of 1) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/ |
| D | cudnn_inplace_fusion.cc | 37 struct AnfNodeIndex { struct
38 AnfNodeIndex() : node(nullptr), index(0) {} in AnfNodeIndex() function
39 AnfNodeIndex(const AnfNodePtr &n, const int i) : node(n), index(i) {} in AnfNodeIndex() argument
101 bool ExistDependencyFromAcc2Cover(const std::vector<AnfNodeIndex> &inplace_node, size_t cover_index… in ExistDependencyFromAcc2Cover()
114 std::pair<size_t, bool> GetCoverIndex(const std::vector<AnfNodeIndex> &inplace_node) { in GetCoverIndex()
160 void CheckInplaceNodeInputs(std::vector<AnfNodeIndex> *inplace_node, size_t cover_index, const Func… in CheckInplaceNodeInputs()
202 void SetNodeAttr(AnfNodeIndex aggregate_node, AnfNodePtr skip_node, std::vector<AnfNodeIndex> *inpl… in SetNodeAttr()
243 bool PatternMatch(const FuncGraphPtr &graph, const AnfNodePtr &node, AnfNodeIndex *aggregate, AnfNo… in PatternMatch()
244 std::vector<AnfNodeIndex> *inplace) { in PatternMatch()
290 inplace->push_back(AnfNodeIndex(inplace_node, inplace_iter->second)); in PatternMatch()
[all …]
|