Searched refs:inplace_node (Results 1 – 1 of 1) sorted by relevance
58 void SetPrimAttr(AnfNodePtr inplace_node, const string &key, const T &value) { in SetPrimAttr() argument59 auto primitive = AnfAlgo::GetCNodePrimitive(inplace_node); in SetPrimAttr()101 bool ExistDependencyFromAcc2Cover(const std::vector<AnfNodeIndex> &inplace_node, size_t cover_index… in ExistDependencyFromAcc2Cover() argument102 if (inplace_node.size() != inplace_node_size) { in ExistDependencyFromAcc2Cover()107 const CNodePtr &cover_node = inplace_node[cover_index].node->cast<CNodePtr>(); in ExistDependencyFromAcc2Cover()108 const CNodePtr &acc_node = inplace_node[acc_index].node->cast<CNodePtr>(); in ExistDependencyFromAcc2Cover()114 std::pair<size_t, bool> GetCoverIndex(const std::vector<AnfNodeIndex> &inplace_node) { in GetCoverIndex() argument115 if (inplace_node.size() != inplace_node_size) { in GetCoverIndex()118 auto first_node = inplace_node[0].node; in GetCoverIndex()119 auto second_node = inplace_node[1].node; in GetCoverIndex()[all …]