Home
last modified time | relevance | path

Searched defs:ret_shape (Results 1 – 13 of 13) sorted by relevance

/third_party/mindspore/mindspore/lite/test/ut/src/api/
Dtensor_c_test.cc33 const int64_t *ret_shape = MSTensorGetShape(tensor, &ret_shape_num); in TEST_F() local
/third_party/mindspore/mindspore/lite/test/ohos_test/src/
Dtensor_test.cc43 const int64_t *ret_shape = OH_AI_TensorGetShape(tensor, &ret_shape_num); variable
/third_party/mindspore/mindspore/core/ops/
Dconcat.cc73 auto ret_shape = element0_shape; in ConcatInfer() local
Dgetnext.cc78 auto ret_shape = !min_shape.empty() && !max_shape.empty() && is_dynamic in GetnextInferShape() local
Dmat_mul.cc78 ShapeVector ret_shape; in MatMulInferShape() local
Dbatch_matmul.cc92 ShapeVector ret_shape; in BatchMatmulInferShape() local
/third_party/mindspore/mindspore/core/abstract/
Dprim_maths.cc305 ShapeVector ret_shape; in InferImplMatMul() local
373 ShapeVector ret_shape; in InferImplBatchMatMul() local
Dutils.cc359 auto ret_shape = std::make_shared<abstract::Shape>(ret_vec, min_shape_vec, max_shape_vec); in MakeAbstractTensor() local
Dprim_nn.cc393 ShapePtr ret_shape = std::make_shared<Shape>(bias_dims); in InferImplBiasAddGrad() local
Dprim_arrays.cc128 auto ret_shape = ret_shape_ptr->shape(); in InferImplStack() local
/third_party/mindspore/mindspore/core/ops/grad/
Dconv2d_backprop_filter.cc57 abstract::ShapePtr ret_shape; in Conv2DBackpropFilterInferShape() local
Dconv2d_backprop_input.cc87 abstract::ShapePtr ret_shape; in Conv2DBackpropInputInferShape() local
/third_party/mindspore/mindspore/ccsrc/utils/
Dconvert_utils_py.cc427 auto ret_shape = std::make_shared<abstract::Shape>(ret_vec, min_shape_vec, max_shape_vec); in MakePyInferRes2AbstractTensor() local