| /third_party/mindspore/mindspore/ccsrc/pipeline/jit/static_analysis/ |
| D | prim.h | 36 : TrivialPrimEvaluator("StandardPrimEvaluator"), prim_(primitive), eval_impl_(eval_impl) {} in StandardPrimEvaluator()
40 PrimitivePtr prim() { return prim_; } in prim()
42 std::string ToString() const override { return identifier_ + prim_->name(); } in ToString()
48 PrimitivePtr prim_; variable
71 …ureEvaluator(const PrimitivePtr primitive) : Evaluator("DoSignatureEvaluator"), prim_(primitive) {} in DoSignatureEvaluator()
81 PrimitivePtr prim_;
86 …aphEvaluator(const PrimitivePtr primitive) : Evaluator("UnpackGraphEvaluator"), prim_(primitive) {} in UnpackGraphEvaluator()
96 PrimitivePtr prim_;
102 : Evaluator("MixedPrecisionCastEvaluator"), prim_(primitive) {} in MixedPrecisionCastEvaluator()
112 PrimitivePtr prim_;
|
| D | prim.cc | 64 auto do_signature = prim_->cast<prim::DoSignaturePrimitivePtr>(); in Run()
99 …new_node = prim::GenerateCNode(out_node->func_graph(), prim_->ToString(), func, args_spec_list, ar… in Run()
101 …new_node = prim::GenerateCNode(out_node->func_graph(), prim_->ToString(), func, args_spec_list, ar… in Run()
151 auto unpack_graph = prim_->cast<prim::UnpackGraphPrimitivePtr>(); in Run()
578 auto prim_py = dyn_cast<PrimitivePy>(prim_); in RunPyInferValue()
586 auto added_attrs = prim_->evaluate_added_attrs(); in RunPyInferValue()
612 auto prim_py = dyn_cast<PrimitivePy>(prim_); in EvalPyCheckPrim()
621 prim_->BeginRecordAddAttr(); in EvalPyCheckPrim()
622 AbstractBasePtr abs_base = eval_impl_.infer_shape_impl_(engine, prim_, args); in EvalPyCheckPrim()
623 prim_->EndRecordAddAttr(); in EvalPyCheckPrim()
[all …]
|
| /third_party/mindspore/mindspore/core/abstract/ |
| D | abstract_function.h | 80 : prim_(prim), tracking_id_(AnfNodeWeakPtr(tracking_id)) {} in prim_() function
84 PrimitivePtr prim() { return prim_; } in MS_DECLARE_PARENT()
90 … Copy() const override { return std::make_shared<PrimitiveAbstractClosure>(prim_, tracking_id()); } in Copy()
95 std::string ToString() const override { return "Prim: " + prim_->name(); } in ToString()
97 ValuePtr RealBuildValue() const override { return prim_; } in RealBuildValue()
100 PrimitivePtr prim_;
269 : prim_(prim), args_spec_list_(args_spec_list), output_(output_spec) {} in TypedPrimitiveAbstractClosure()
273 PrimitivePtr prim() { return prim_; } in MS_DECLARE_PARENT()
277 return std::make_shared<TypedPrimitiveAbstractClosure>(prim_, args_spec_list_, output_); in Copy()
285 PrimitivePtr prim_;
|
| D | abstract_function.cc | 142 MS_EXCEPTION_IF_NULL(prim_); in operator ==()
143 return (prim_ == other_prim->prim_ && tracking_id() == other_prim->tracking_id()); in operator ==()
147 auto hash_value = hash_combine(tid(), prim_->hash()); in hash()
149 hash_value = hash_combine(hash_value, std::hash<Primitive *>{}(prim_.get())); in hash()
294 if (prim_ != other_typed->prim_) { in operator ==()
304 auto hash_value = hash_combine(tid(), prim_->hash()); in hash()
311 buffer << "TypedPrimitiveAbstractClosure: primitive: " << prim_->name() << "(args: {"; in ToString()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/ |
| D | prim_eliminate.h | 30 explicit PrimEliminater(const PrimitivePtr &prim) : prim_(prim) {} in PrimEliminater()
35 AnfVisitor::Match(prim_, {IsNode})(node); in operator() local
43 PrimitivePtr prim_; variable
|
| D | reshape_eliminate.h | 96 auto new_node = fg->NewCNode({NewValueNode(prim_), x_, shape_}); in operator()
113 prim_ = GetValueNode<PrimitivePtr>(inputs[0]); in Visit()
121 prim_ = nullptr; in Reset()
127 PrimitivePtr prim_{nullptr};
|
| /third_party/mindspore/mindspore/core/ir/ |
| D | func_graph.h | 108 : prim_(prim), func_graph_(FuncGraphWeakPtr(func_graph)) {} in prim_() function
111 : prim_(prim), func_graph_(FuncGraphWeakPtr(func_graph)) {} in prim_() function
113 FuncGraphTransform(const FuncGraphTransform &t) : prim_(t.prim_), func_graph_(t.func_graph_) {} in FuncGraphTransform()
128 PrimitivePtr primitive() const { return prim_; } in primitive()
132 prim_ = t.prim_;
139 PrimitivePtr prim_;
|
| D | pattern_matcher.h | 95 : prim_(prim), x_(x), y_(y), is_commutative_(is_commutative) {} in prim_() function
101 AnfNodePtrList list = {NewValueNode(prim_), lhs, rhs}; in GetNode()
106 if (IsPrimitiveCNode(node, prim_)) { in TryCapture_()
149 const PrimitivePtr prim_;
159 PUnaryOperation(const PrimitivePtr &prim, const T &x) : prim_(prim), x_(x) {} in PUnaryOperation()
163 AnfNodePtrList list = {NewValueNode(prim_), x_.GetNode(node)}; in GetNode()
168 if (IsPrimitiveCNode(node, prim_)) { in TryCapture_()
194 const PrimitivePtr prim_;
349 …explicit PPrimitive(const PrimitivePtr &prim, const TArgs &... args) : prim_(prim), args_(args...)…
356 prim_cnode.insert(prim_cnode.begin(), NewValueNode(prim_));
[all …]
|
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/ |
| D | pattern.h | 134 prim_ = std::make_shared<PrimitivePy>(prim_obj, prim_adapter); in Call()
137 prim_ = std::make_shared<PrimitivePy>(prim_name); in Call()
141 unique_name_ = std::to_string(g_id_++) + "Call_" + prim_->ToString(); in Call()
146 PrimitivePtr prim_value() { return prim_; } in prim_value()
151 PrimitivePtr prim_ = nullptr; variable
|
| D | pattern.cc | 56 } else if (prim_ != nullptr) { in match()
58 if (!IsPrimitive(cnode->input(0), prim_)) { in match()
|