Searched refs:origin_prim (Results 1 – 3 of 3) sorted by relevance
124 auto origin_prim = GetValueNode<PrimitivePtr>(cnode->input(0)); in ClonePrimitive() local125 MS_ASSERT(origin_prim != nullptr); in ClonePrimitive()128 if (op_primc_fns.find(origin_prim->name()) != op_primc_fns.end()) { in ClonePrimitive()129 prim = op_primc_fns[origin_prim->name()](); in ClonePrimitive()131 prim = std::make_shared<PrimitiveC>(origin_prim->name()); in ClonePrimitive()133 prim->set_instance_name(origin_prim->name()); in ClonePrimitive()135 prim->SetAttrs(origin_prim->attrs()); in ClonePrimitive()
290 PrimitivePtr origin_prim = GetValueNode<PrimitivePtr>(node->input(0)); in ReplaceOneOp() local291 SetUserAttrs(origin_prim->attrs(), prim); in ReplaceOneOp()
749 PrimitivePtr origin_prim = GetValueNode<PrimitivePtr>(node->input(0)); in StepReplaceOp() local750 SetUserAttrs(origin_prim->attrs(), prim); in StepReplaceOp()751 auto origin_prim_attrs = origin_prim->attrs(); in StepReplaceOp()