Searched refs:pre_cnode (Results 1 – 7 of 7) sorted by relevance
59 auto pre_cnode = GetRealPrevCNode(cnode->input(1)); in ConcatSplitEliminate() local60 if (pre_cnode == nullptr || !CheckPrimitiveType(pre_cnode, prim::kPrimConcat)) { in ConcatSplitEliminate()65 auto finder = graph_node_outputs.find(pre_cnode->fullname_with_scope()); in ConcatSplitEliminate()73 size_t pre_inputs_size = pre_cnode->inputs().size(); in ConcatSplitEliminate()75 auto pre_prim = GetValueNode<std::shared_ptr<ops::Concat>>(pre_cnode->input(kAnfPrimitiveIndex)); in ConcatSplitEliminate()129 if (!manager->Replace((inputs_node[i - 1])->input(1), pre_cnode->input(i))) { in ConcatSplitEliminate()
127 auto pre_cnode = cnode->input(1)->cast<CNodePtr>(); in TransTransFusion() local128 MS_ASSERT(pre_cnode != nullptr); in TransTransFusion()129 if (GetTransposePerm(pre_cnode, &pre_perm) != lite::RET_OK) { in TransTransFusion()130 MS_LOG(ERROR) << "transpose rm cannot be obtained, " << pre_cnode->fullname_with_scope(); in TransTransFusion()134 if (!manager_->Replace(cnode, pre_cnode->input(1))) { in TransTransFusion()
159 auto pre_cnode = pre_node->cast<CNodePtr>(); in TransTransFusion() local160 if (pre_cnode == nullptr) { in TransTransFusion()163 if (IsMarkedTrainOp(pre_cnode)) { in TransTransFusion()166 if (GetTransposePerm(pre_cnode, &pre_perm) != lite::RET_OK) { in TransTransFusion()171 return pre_cnode->input(1); in TransTransFusion()
1799 CNodePtr pre_cnode = node->cast<CNodePtr>(); in FindPreNodes() local1800 if (!IsValueNode<Primitive>(pre_cnode->input(0))) { in FindPreNodes()1804 for (size_t index = 1; index < pre_cnode->inputs().size(); ++index) { in FindPreNodes()1805 auto next_node = pre_cnode->inputs()[index]; in FindPreNodes()1816 unique_ids->push_back(pre_cnode->UniqueId()); in FindPreNodes()2257 auto pre_cnode = pre_node->cast<CNodePtr>(); in FindLossCNode() local2258 pre_cnode = HandleDependLoss(pre_cnode, 0); in FindLossCNode()2259 if (pre_cnode->input(0)->isa<CNode>()) { in FindLossCNode()2260 auto switch_cnode = pre_cnode->input(0)->cast<CNodePtr>(); in FindLossCNode()2267 if (pre_cnode == nullptr || !IsValueNode<Primitive>(pre_cnode->input(0))) { in FindLossCNode()[all …]
337 CNodePtr pre_cnode = pre_node->cast<CNodePtr>(); in FindReshapePreNodeStraCosts() local338 auto pre_op_info = pre_cnode->user_data<OperatorInfo>(); in FindReshapePreNodeStraCosts()339 if (IsParallelCareNode(pre_cnode) && (pre_op_info != nullptr)) { in FindReshapePreNodeStraCosts()
68 auto pre_cnode = pre_node->cast<CNodePtr>(); in FindAccuGrad() local69 pre_node = pre_cnode->input(1); in FindAccuGrad()
814 auto pre_cnode = kernels[IntToSize(k)]; in GenKernelEvents() local815 auto pre_cnode_stream_id = AnfAlgo::GetStreamId(pre_cnode); in GenKernelEvents()821 if (visited == pre_cnode && !stream_hit[pre_cnode_stream_id]) { in GenKernelEvents()