Searched defs:node_prim (Results 1 – 14 of 14) sorted by relevance
64 …auto node_prim = inner::OpRegistry::Instance().NewOp(GetValueNode<PrimitivePtr>(cnode->input(0))->… in ElemwiseBroadcastChecker() local92 …auto node_prim = inner::OpRegistry::Instance().NewOp(GetValueNode<PrimitivePtr>(cnode->input(0))->… in ReduceChecker() local177 …auto node_prim = inner::OpRegistry::Instance().NewOp(GetValueNode<PrimitivePtr>(cnode->input(0))->… in FindInputOutputPairs() local
40 auto node_prim = GetValueNode<PrimitivePtr>(mirror_cnode->input(0)); in SetMirrorFusion() local63 auto node_prim = GetValueNode<PrimitivePtr>(comm_cnode->input(0)); in AdjustRelatedFusionNode() local
39 PrimitivePtr node_prim = prim_node_anf->value()->cast<PrimitivePtr>(); in FindPrimtive() local
500 PrimitivePtr node_prim = prim_anf_node->value()->cast<PrimitivePtr>(); in FindReshapeNextNodeStraCosts() local
63 auto node_prim = GetCNodePrimitive(cnode); in FindTargetNodes() local
312 auto node_prim = GetCNodePrimitive(node); in Process() local
143 auto node_prim = GetValueNode<PrimitivePtr>(fn); in GetPrimitiveName() local
55 auto node_prim = std::make_shared<ops::LayerNormFusion>()->GetPrim(); in LayerNormFusionInferShape() local
545 auto node_prim = GetValueNode<PrimitivePtr>(fn); in IsSubGraph() local817 auto node_prim = GetValueNode<PrimitivePtr>(fn); in IsCut() local
55 PrimitivePtr node_prim = prim_anf_node->value()->cast<PrimitivePtr>(); in GetCNodeOpName() local
1607 PrimitivePtr node_prim = prim_anf_node->value()->cast<PrimitivePtr>(); in FindParameterNextLayout() local
2677 PrimitivePtr node_prim = prim_anf_node->value()->cast<PrimitivePtr>(); in FindCNode() local
1354 PrimitivePtr node_prim = prim_node_anf->value()->cast<PrimitivePtr>(); in FindParallelCareNode() local
121 PrimitivePtr node_prim = GetCNodePrimitive(node); in get_node_prim_name() local