Searched refs:infer_shape_impl_ (Results 1 – 7 of 7) sorted by relevance
| /third_party/mindspore/tests/ut/cpp/abstract/ |
| D | abstract_test.cc | 97 ASSERT_TRUE(nullptr != eval_impl.infer_shape_impl_); in TEST_F()
99 AbstractBasePtr new_cls = eval_impl.infer_shape_impl_(nullptr, prim::kPrimMakeRecord, args_list); in TEST_F()
|
| /third_party/mindspore/mindspore/core/ops/ |
| D | primitive_c.cc | 35 auto infer_function = iter->second.infer_shape_impl_; in Infer()
|
| /third_party/mindspore/mindspore/core/abstract/ |
| D | primitive_infer_map.h | 36 InferShapeImpl infer_shape_impl_; // infer shape and type for ops member
|
| /third_party/mindspore/mindspore/core/load_mindir/ |
| D | infer_mindir.cc | 157 MS_EXCEPTION_IF_NULL(ret->second.infer_shape_impl_); in InferPrimitiveShape()
158 return ret->second.infer_shape_impl_(nullptr, prim, args_spec_list); in InferPrimitiveShape()
164 MS_EXCEPTION_IF_NULL(ret_backend->second.infer_shape_impl_); in InferPrimitiveShape()
165 return ret_backend->second.infer_shape_impl_(nullptr, prim, args_spec_list); in InferPrimitiveShape()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/common/ |
| D | helper.cc | 944 MS_EXCEPTION_IF_NULL(ret->second.infer_shape_impl_); in CppInferShape()
946 return ret->second.infer_shape_impl_(nullptr, prim, infer_spec_list); in CppInferShape()
952 MS_EXCEPTION_IF_NULL(ret_backend->second.infer_shape_impl_); in CppInferShape()
957 return ret_backend->second.infer_shape_impl_(nullptr, prim, infer_spec_list); in CppInferShape()
|
| /third_party/mindspore/mindspore/ccsrc/pipeline/jit/static_analysis/ |
| D | prim.cc | 622 AbstractBasePtr abs_base = eval_impl_.infer_shape_impl_(engine, prim_, args); in EvalPyCheckPrim()
668 abs_base = eval_impl_.infer_shape_impl_(engine, prim_, args); in EvalPrim()
|
| D | static_analysis.cc | 377 if (eval_impl.infer_shape_impl_ != nullptr) { in GetPrimEvaluator()
|