Searched refs:kPrimGetRefValue (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/ |
| D | ref_eliminate.h | 45 new_fg->set_output(new_fg->NewCNode({NewValueNode(prim::kPrimGetRefValue), output})); in operator()
68 MATCH_REPLACE(node, PPrimitive(prim::kPrimGetRefValue, x), x); in operator()
81 …MATCH_REPLACE(node, PPrimitive(prim::kPrimGetRefValue, PPrimitive(prim::kPrimMakeRef, x, y, z)), y… in operator()
82 …MATCH_REPLACE_IF(node, PPrimitive(prim::kPrimGetRefValue, x), x, x.CheckFunc(IsCNodeSwitch, node)); in operator()
91 …MATCH_REPLACE_LAMBDA_IF(node, PPrimitive(prim::kPrimGetRefValue, x), GetRefLambda, x.CheckFunc(IsC… in operator()
|
| D | special_op_eliminate.h | 48 get_ref_value_(std::make_shared<PrimEliminater>(prim::kPrimGetRefValue)), in SpecialOpEliminater()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/ |
| D | irpass.cc | 66 … prim::kPrimPrintShapeType, prim::kPrimGetRefValue, prim::kPrimMirror, prim::kPrimVirtualDiv}); in OptimizeIRPassLib()
144 …on(std::make_shared<GetRefParamEliminater>(), "get_ref_param_eliminate", {prim::kPrimGetRefValue}); in OptimizeIRPassLib()
146 {prim::kPrimGetRefKey, prim::kPrimGetRefValue}); in OptimizeIRPassLib()
|
| /third_party/mindspore/mindspore/core/abstract/ |
| D | primitive_infer_map.cc | 180 {prim::kPrimGetRefValue, {InferImplGetRefValue, nullptr, true}}, in GetPrimitiveToEvalImplMap()
|
| /third_party/mindspore/mindspore/core/base/ |
| D | core_ops.h | 619 inline const PrimitivePtr kPrimGetRefValue = std::make_shared<Primitive>("get_ref_value"); variable
|