Searched refs:graph_cnode (Results 1 – 3 of 3) sorted by relevance
83 bool LabelParameterStart(const FuncGraphPtr &graph, const CNodePtr &graph_cnode);84 bool NeedGrad(const CNodePtr &cnode, const CNodePtr &graph_cnode);
91 bool PipelineTransformer::NeedGrad(const CNodePtr &cnode, const CNodePtr &graph_cnode) { in NeedGrad() argument107 MS_EXCEPTION_IF_NULL(graph_cnode); in NeedGrad()108 auto real_param = graph_cnode->input(LongToSize(pos + 1)); in NeedGrad()118 … PipelineTransformer::LabelParameterStart(const FuncGraphPtr &graph, const CNodePtr &graph_cnode) { in LabelParameterStart() argument137 if (NeedGrad(cnode, graph_cnode)) { in LabelParameterStart()
110 auto graph_cnode = utils::cast<CNodePtr>(node); in MarkTrainWeightSharingOp() local111 MS_CHECK_TRUE_RET(graph_cnode != nullptr, RET_ERROR); in MarkTrainWeightSharingOp()112 auto graph_prim = GetValueNode<PrimitivePtr>(graph_cnode->input(0)); in MarkTrainWeightSharingOp()117 for (size_t i = 1; i < graph_cnode->inputs().size(); i++) { in MarkTrainWeightSharingOp()119 if ((graph_cnode->input(i) == cnode->input(j)) && utils::isa<Parameter>(cnode->input(j))) { in MarkTrainWeightSharingOp()