| /third_party/mindspore/mindspore/core/ops/ |
| D | reshape.cc | 62 ShapeVector x_min_shape = x->shape()->min_shape(); in ReshapeInfer() local
66 if (x_min_shape.empty()) { in ReshapeInfer()
67 x_min_shape = x_shape; in ReshapeInfer()
72 …std::accumulate(x_min_shape.begin(), x_min_shape.end(), static_cast<int64_t>(1), std::multiplies<i… in ReshapeInfer()
|
| D | op_utils.cc | 65 auto x_min_shape = x_shape_map[kMinShape]; in BroadCastInferShape() local
70 return std::make_shared<abstract::Shape>(x_shape, x_min_shape, x_max_shape); in BroadCastInferShape()
73 auto min_broadcast_shape = CalBroadCastShape(x_min_shape, y_min_shape, op_name); in BroadCastInferShape()
|
| D | transpose.cc | 32 …auto x_min_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[0]->BuildShape())[kM… in InferShape() local
71 if (!x_min_shape.empty() && !x_max_shape.empty()) { in InferShape()
75 min_shape.push_back(x_min_shape[LongToSize(i)]); in InferShape()
|
| D | conv2d.cc | 159 auto x_min_shape = x_shape_map[kMinShape]; in Conv2dInferShape() local
163 CheckAndConvertUtils::CheckMinMaxShape(x_shape, &x_min_shape, &x_max_shape); in Conv2dInferShape()
167 CheckShapeAllPositive(prim_name + " x_min_shape", x_min_shape); in Conv2dInferShape()
216 …Conv2DPadFunction(&output_hw_min, &pad_list_min, x_min_shape[h_axis], x_min_shape[w_axis], kernel_… in Conv2dInferShape()
228 output_shape_min = {x_min_shape[n_axis], output_hw_min[0], output_hw_min[1], out_channel}; in Conv2dInferShape()
232 output_shape_min = {x_min_shape[n_axis], out_channel, output_hw_min[0], output_hw_min[1]}; in Conv2dInferShape()
|
| D | mat_mul.cc | 58 ShapeVector x_min_shape = x_shape_map[kMinShape]; in MatMulInferShape() local
62 CheckAndConvertUtils::CheckMinMaxShape(x_shp, &x_min_shape, &x_max_shape); in MatMulInferShape()
88 make_shape(ret_min_shape, x_min_shape, y_min_shape); in MatMulInferShape()
|
| D | batch_matmul.cc | 70 ShapeVector x_min_shape = x_shape_map[kMinShape]; in BatchMatmulInferShape() local
74 CheckAndConvertUtils::CheckMinMaxShape(x_shp, &x_min_shape, &x_max_shape); in BatchMatmulInferShape()
114 make_shape(ret_min_shape, x_min_shape, y_min_shape); in BatchMatmulInferShape()
|
| /third_party/mindspore/mindspore/core/abstract/ |
| D | prim_nn.cc | 137 ShapeVector x_min_shape = input_x->shape()->min_shape(); in InferImplBatchNorm() local
139 CheckMinMaxShape(x_shape, &x_min_shape, &x_max_shape); in InferImplBatchNorm()
177 ShapePtr output_shape_ptr = std::make_shared<Shape>(x_shape, x_min_shape, x_max_shape); in InferImplBatchNorm()
252 ShapeVector x_min_shape = input_x->shape()->min_shape(); in InferImplConv2D() local
254 CheckMinMaxShape(x_shape, &x_min_shape, &x_max_shape); in InferImplConv2D()
256 CheckShapeAllPositive(op_name + " x_min_shape", x_min_shape); in InferImplConv2D()
314 …Conv2DPadFunction(&output_hw_min, &pad_list_min, x_min_shape[h_axis], x_min_shape[w_axis], kernel_… in InferImplConv2D()
326 output_shape_min = {x_min_shape[n_axis], output_hw_min[0], output_hw_min[1], out_channel}; in InferImplConv2D()
330 output_shape_min = {x_min_shape[n_axis], out_channel, output_hw_min[0], output_hw_min[1]}; in InferImplConv2D()
357 ShapeVector x_min_shape = x->shape()->min_shape(); in InferImplBiasAdd() local
[all …]
|
| D | prim_maths.cc | 287 ShapeVector x_min_shape = x->shape()->min_shape(); in InferImplMatMul() local
291 CheckMinMaxShape(x_shp, &x_min_shape, &x_max_shape); in InferImplMatMul()
315 make_shape(ret_min_shape, x_min_shape, y_min_shape); in InferImplMatMul()
354 ShapeVector x_min_shape = x->shape()->min_shape(); in InferImplBatchMatMul() local
358 CheckMinMaxShape(x_shp, &x_min_shape, &x_max_shape); in InferImplBatchMatMul()
395 make_shape(ret_min_shape, x_min_shape, y_min_shape); in InferImplBatchMatMul()
|
| D | prim_arrays.cc | 759 ShapeVector x_min_shape = x->shape()->min_shape(); in InferImplReshape() local
763 if (x_min_shape.empty()) { in InferImplReshape()
764 x_min_shape = x_shape; in InferImplReshape()
783 for (int64_t value : x_min_shape) { in InferImplReshape()
1094 ShapeVector x_min_shape = x->shape()->min_shape(); in InferImplArgMaxWithValue() local
1096 CheckMinMaxShape(x_shape, &x_min_shape, &x_max_shape); in InferImplArgMaxWithValue()
1098 cal_shape(min_shape, x_min_shape); in InferImplArgMaxWithValue()
|
| /third_party/mindspore/mindspore/ops/operations/ |
| D | array_ops.py | 507 x_min_shape = x['min_shape']
509 x_min_shape = x['shape']
511 min_arr_prod = np.prod(x_min_shape)
|