Home
last modified time | relevance | path

Searched refs:impl_reg (Results 1 – 2 of 2) sorted by relevance

/third_party/mindspore/mindspore/core/abstract/
Dprimitive_infer_map.h54 …sterStandardPrimitiveImpl(const PrimitivePtr &primitive, const StandardPrimitiveImplReg &impl_reg);
60 const StandardPrimitiveImplReg impl_reg{infer_impl, infer_value_impl, is_wight_list};
61 RegisterStandardPrimitiveImpl(primitive, impl_reg);
Dprimitive_infer_map.cc265 …terStandardPrimitiveImpl(const PrimitivePtr &primitive, const StandardPrimitiveImplReg &impl_reg) { in RegisterStandardPrimitiveImpl() argument
267 prim_eval_map[primitive] = impl_reg; in RegisterStandardPrimitiveImpl()