Home
last modified time | relevance | path

Searched refs:CheckMinMaxShape (Results 1 – 11 of 11) sorted by relevance

/third_party/mindspore/mindspore/core/ops/
Dmat_mul.cc62 CheckAndConvertUtils::CheckMinMaxShape(x_shp, &x_min_shape, &x_max_shape); in MatMulInferShape()
63 CheckAndConvertUtils::CheckMinMaxShape(y_shp, &y_min_shape, &y_max_shape); in MatMulInferShape()
Dbatch_matmul.cc74 CheckAndConvertUtils::CheckMinMaxShape(x_shp, &x_min_shape, &x_max_shape); in BatchMatmulInferShape()
75 CheckAndConvertUtils::CheckMinMaxShape(y_shp, &y_min_shape, &y_max_shape); in BatchMatmulInferShape()
Dbias_add.cc70 CheckAndConvertUtils::CheckMinMaxShape(input_shape, &min_shape, &max_shape); in InferShape()
Dconv2d.cc163 CheckAndConvertUtils::CheckMinMaxShape(x_shape, &x_min_shape, &x_max_shape); in Conv2dInferShape()
164 CheckAndConvertUtils::CheckMinMaxShape(w_shape, &w_min_shape, &w_max_shape); in Conv2dInferShape()
/third_party/mindspore/mindspore/core/abstract/
Dprim_nn.cc139 CheckMinMaxShape(x_shape, &x_min_shape, &x_max_shape); in InferImplBatchNorm()
176 CheckMinMaxShape(gamma_shape, &gamma_min_shape, &gamma_max_shape); in InferImplBatchNorm()
237 CheckMinMaxShape(w_shape, &w_min_shape, &w_max_shape); in CheckShape()
254 CheckMinMaxShape(x_shape, &x_min_shape, &x_max_shape); in InferImplConv2D()
372 CheckMinMaxShape(x_shape, &x_min_shape, &x_max_shape); in InferImplBiasAdd()
443 CheckMinMaxShape(shape, &min_shape, &max_shape); in InferImplDropout()
Dutils.h61 void CheckMinMaxShape(const ShapeVector &shape, ShapeVector *min_shape, ShapeVector *max_shape);
Dprim_maths.cc291 CheckMinMaxShape(x_shp, &x_min_shape, &x_max_shape); in InferImplMatMul()
292 CheckMinMaxShape(y_shp, &y_min_shape, &y_max_shape); in InferImplMatMul()
358 CheckMinMaxShape(x_shp, &x_min_shape, &x_max_shape); in InferImplBatchMatMul()
359 CheckMinMaxShape(y_shp, &y_min_shape, &y_max_shape); in InferImplBatchMatMul()
Dprim_arrays.cc396 CheckMinMaxShape(shape, &min_shape, &max_shape); in InferImplScatterAdd()
411 CheckMinMaxShape(shape, &min_shape, &max_shape); in InferImplScatterSub()
425 CheckMinMaxShape(shape, &min_shape, &max_shape); in InferImplScatterUpdate()
740 CheckMinMaxShape(input_shp, &x_min_shp, &x_max_shp); in InferImplTranspose()
953 CheckMinMaxShape(shape_base, &min_shape_base, &max_shape_base); in InferImplConcat()
978 CheckMinMaxShape(shape_tensor, &min_shape_tensor, &max_shape_tensor); in InferImplConcat()
1002 CheckMinMaxShape(shape, &min_shape, &max_shape); in InferImplConcat()
1096 CheckMinMaxShape(x_shape, &x_min_shape, &x_max_shape); in InferImplArgMaxWithValue()
Dutils.cc311 void CheckMinMaxShape(const ShapeVector &shape, ShapeVector *min_shape, ShapeVector *max_shape) { in CheckMinMaxShape() function
/third_party/mindspore/mindspore/core/utils/
Dcheck_convert_utils.h311 …static void CheckMinMaxShape(const ShapeVector &shape, ShapeVector *min_shape, ShapeVector *max_sh…
Dcheck_convert_utils.cc736 void CheckAndConvertUtils::CheckMinMaxShape(const ShapeVector &shape, ShapeVector *min_shape, Shape… in CheckMinMaxShape() function in mindspore::CheckAndConvertUtils