Searched refs:kPrimMakeRef (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/ |
| D | ref_eliminate.h | 58 MATCH_REPLACE(node, PPrimitive(prim::kPrimMakeRef, x, y, z), y); in operator()
80 … MATCH_REPLACE(node, PPrimitive(prim::kPrimGetRefKey, PPrimitive(prim::kPrimMakeRef, x, y, z)), x); in operator()
81 …MATCH_REPLACE(node, PPrimitive(prim::kPrimGetRefValue, PPrimitive(prim::kPrimMakeRef, x, y, z)), y… in operator()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/ |
| D | irpass.cc | 142 … MakeSubstitution(std::make_shared<MakeRefEliminater>(), "make_ref_eliminate", prim::kPrimMakeRef); in OptimizeIRPassLib()
|
| /third_party/mindspore/mindspore/core/abstract/ |
| D | primitive_infer_map.cc | 178 {prim::kPrimMakeRef, {InferImplMakeRef, nullptr, true}}, in GetPrimitiveToEvalImplMap()
|
| /third_party/mindspore/mindspore/core/base/ |
| D | core_ops.h | 618 inline const PrimitivePtr kPrimMakeRef = std::make_shared<Primitive>("make_ref"); variable
|