Searched defs:inplace_node (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()
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 …odeAttr(AnfNodeIndex aggregate_node, AnfNodePtr skip_node, std::vector<AnfNodeIndex> *inplace_node, in SetNodeAttr()
271 auto inplace_node = AnfAlgo::GetInputNode(utils::cast<CNodePtr>(*skip_node), i); in PatternMatch() local
313 std::vector<AnfNodeIndex> inplace_node; in Run() local
|