Home
last modified time | relevance | path

Searched refs:TypedPrimitiveAbstractClosure (Results 1 – 6 of 6) sorted by relevance

/third_party/mindspore/mindspore/core/abstract/
Dabstract_function.cc286 bool TypedPrimitiveAbstractClosure::operator==(const AbstractFunction &other) const { in operator ==()
287 if (!other.isa<TypedPrimitiveAbstractClosure>()) { in operator ==()
290 auto other_typed = static_cast<const TypedPrimitiveAbstractClosure *>(&other); in operator ==()
303 std::size_t TypedPrimitiveAbstractClosure::hash() const { in hash()
309 std::string TypedPrimitiveAbstractClosure::ToString() const { in ToString()
Dabstract_function.h262 class MS_CORE_API TypedPrimitiveAbstractClosure : public AbstractFuncAtom {
267 TypedPrimitiveAbstractClosure(const PrimitivePtr prim, const AbstractBasePtrList &args_spec_list, in TypedPrimitiveAbstractClosure() function
270 ~TypedPrimitiveAbstractClosure() override = default;
271 MS_DECLARE_PARENT(TypedPrimitiveAbstractClosure, AbstractFuncAtom) in MS_DECLARE_PARENT() argument
277 return std::make_shared<TypedPrimitiveAbstractClosure>(prim_, args_spec_list_, output_); in Copy()
/third_party/mindspore/mindspore/ccsrc/vm/
Dtransform.cc45 using TypedPrimitiveAbstractClosurePtr = std::shared_ptr<abstract::TypedPrimitiveAbstractClosure>;
473 …TypedPrimitiveAbstractClosurePtr tp = dyn_cast<abstract::TypedPrimitiveAbstractClosure>(type->GetU… in WrapPrimitives()
/third_party/mindspore/mindspore/ccsrc/pipeline/jit/static_analysis/
Dstatic_analysis.cc494 EvaluatorPtr AnalysisEngine::_GetEvaluatorFor(const std::shared_ptr<TypedPrimitiveAbstractClosure> … in _GetEvaluatorFor() argument
513 } else if (func->isa<TypedPrimitiveAbstractClosure>()) { in _GetEvaluatorFor()
514 return _GetEvaluatorFor(func->cast<std::shared_ptr<TypedPrimitiveAbstractClosure>>()); in _GetEvaluatorFor()
Dstatic_analysis.h268 EvaluatorPtr _GetEvaluatorFor(const std::shared_ptr<TypedPrimitiveAbstractClosure> &);
Dprogram_specialize.cc472 auto real_func = dyn_cast<TypedPrimitiveAbstractClosure>(func); in BuildSpecializedNodeInner()
492 …auto type_func = std::make_shared<TypedPrimitiveAbstractClosure>(prim_func->prim(), argvals, uniqu… in BuildSpecializedNodeInner()