Searched refs:cnode_prim (Results 1 – 6 of 6) sorted by relevance
91 CNodePtr cnode_prim = func_graph->NewCNode(inputs); in MakeFuncGraph() local94 inputs.push_back(cnode_prim); in MakeFuncGraph()
147 CNodePtr cnode_prim = anf_graph->NewCNode(inputs); in TEST_F() local151 inputs.push_back(cnode_prim); in TEST_F()736 CNodePtr cnode_prim = func_graph->NewCNode(inputs); in TEST_F() local739 inputs.push_back(cnode_prim); in TEST_F()
159 auto cnode_prim = GetValueNode<PrimitivePtr>(cnode->input(0)); in InferShape() local160 MS_CHECK_TRUE_MSG(cnode_prim != nullptr, lite::RET_NULL_PTR, "GetValueNode Failed"); in InferShape()161 cnode_prim->AddAttr(ops::kFormat, MakeValue<int64_t>(inputs[0]->format())); in InferShape()
78 CNodePtr cnode_prim = func_graph->NewCNode(inputs); in MakeFuncGraph() local81 inputs.push_back(cnode_prim); in MakeFuncGraph()
124 CNodePtr cnode_prim = func_graph->NewCNode(inputs); in MakeFuncGraph() local127 inputs.push_back(cnode_prim); in MakeFuncGraph()
1527 auto cnode_prim = GetValueNode<PrimitivePtr>(cnode->input(0)); in InsertAllGatherOp() local1528 MS_EXCEPTION_IF_NULL(cnode_prim); in InsertAllGatherOp()