Searched refs:in_cnode (Results 1 – 4 of 4) sorted by relevance
335 for (const auto &in_cnode : lite::GetInputCNode(node)) { in FindBeforeSubGraphInBelongAnf() local336 if (in_cnode == nullptr) { in FindBeforeSubGraphInBelongAnf()339 q.push(in_cnode); in FindBeforeSubGraphInBelongAnf()347 for (const auto &in_cnode : lite::GetInputCNode(cur_cnode)) { in FindBeforeSubGraphInBelongAnf() local348 q.push(in_cnode); in FindBeforeSubGraphInBelongAnf()387 for (const auto &in_cnode : lite::GetInputCNode(cur_cnode)) { in FindAfterSubGraphInBelongAnf() local388 if (subgraph_out_nodes.count(in_cnode) == 0 && before_out_nodes.count(in_cnode) == 0) { in FindAfterSubGraphInBelongAnf()389 q.push(in_cnode); in FindAfterSubGraphInBelongAnf()484 for (const auto &in_cnode : this->GetInCNodes()) { in CreateParameterForPartialSubGraph()485 if (in_cnode == nullptr) { in CreateParameterForPartialSubGraph()[all …]
104 auto in_cnode = in_node->cast<CNodePtr>(); in InputCheck() local105 MS_EXCEPTION_IF_NULL(in_cnode); in InputCheck()106 auto in_node_name = AnfAlgo::GetCNodeName(in_cnode); in InputCheck()116 …if ((AnfAlgo::HasNodeAttr("non_task", in_cnode) && AnfAlgo::GetNodeAttr<bool>(in_node, "non_task")… in InputCheck()117 opt::IsNopNode(in_cnode)) { in InputCheck()
311 auto in_cnode = in_node->cast<CNodePtr>(); in FindRoots() local312 if (in_cnode != nullptr) { in FindRoots()313 (void)roots->erase(in_cnode); in FindRoots()336 auto in_cnode = in_node->cast<CNodePtr>(); in FindLeaves() local337 if (in_cnode != nullptr) { in FindLeaves()338 if (std::find(segment.begin(), segment.end(), in_cnode) != segment.end()) { in FindLeaves()
420 for (auto &in_cnode : in_nodes) { in HandleGraphMultiNode() local421 if (CheckPrimitiveType(in_cnode, prim::kPrimTranspose)) { in HandleGraphMultiNode()422 visit_transposes->insert(in_cnode); in HandleGraphMultiNode()440 for (auto &in_cnode : in_nodes) { in HandleGraphMultiNode() local441 manager->Replace(in_cnode, in_cnode->input(1)); in HandleGraphMultiNode()