Searched refs:infer_impl (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/mindspore/core/abstract/ |
| D | primitive_infer_map.h | 58 …egisterStandardPrimitiveEvalHelper(const PrimitivePtr &primitive, const InferShapeImpl &infer_impl,
60 const StandardPrimitiveImplReg impl_reg{infer_impl, infer_value_impl, is_wight_list};
66 #define REGISTER_PRIMITIVE_EVAL_IMPL(name, primitive, infer_impl, infer_value_impl, is_wight_list) … argument
68 …abstract::RegisterStandardPrimitiveEvalHelper(primitive, infer_impl, infer_value_impl, is_wight_li…
|
| /third_party/mindspore/mindspore/ccsrc/frontend/operator/ |
| D | ops_front_infer_function.h | 62 #define REGISTER_PRIMITIVE_FRONT_EVAL_IMPL(name, primitive, infer_impl, infer_value_impl) \ argument
63 …auto helper_##name = abstract::RegisterStandardPrimitiveEvalHelper(primitive, infer_impl, infer_va…
|