Home
last modified time | relevance | path

Searched refs:CppInferShape (Results 1 – 4 of 4) sorted by relevance

/third_party/mindspore/tests/ut/cpp/pre_activate/common/
Drestore_abs_input_in_backed_infer_test.cc83 auto attr_infer_result = CppInferShape(prim_attr_test, {abs_a, abs_c}); in TEST_F()
89 …CppInferShape(prim_dynamic_input_test, {abs_dynamic_a, abs_dynamic_b, abs_dynamic_c, abs_dynamic_d… in TEST_F()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/common/
Dhelper.h222 AbstractBasePtr CppInferShape(const PrimitivePtr &prim, const AbstractBasePtrList &args_spec_list);
Dhelper.cc938 AbstractBasePtr CppInferShape(const PrimitivePtr &prim, const AbstractBasePtrList &args_spec_list) { in CppInferShape() function
/third_party/mindspore/mindspore/ccsrc/backend/session/
Danf_runtime_algorithm.cc2270 auto eval_result = opt::CppInferShape(primitive, args_spec_list); in InferShape()