| /third_party/mindspore/mindspore/core/ops/ |
| D | reshape.cc | 61 ShapeVector x_max_shape = x->shape()->max_shape(); in ReshapeInfer() local
63 if (x_max_shape.empty()) { in ReshapeInfer()
64 x_max_shape = x_shape; in ReshapeInfer()
70 …std::accumulate(x_max_shape.begin(), x_max_shape.end(), static_cast<int64_t>(1), std::multiplies<i… in ReshapeInfer()
|
| D | op_utils.cc | 66 auto x_max_shape = x_shape_map[kMaxShape]; in BroadCastInferShape() local
70 return std::make_shared<abstract::Shape>(x_shape, x_min_shape, x_max_shape); in BroadCastInferShape()
74 auto max_broadcast_shape = CalBroadCastShape(x_max_shape, y_max_shape, op_name); in BroadCastInferShape()
|
| D | transpose.cc | 33 …auto x_max_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[0]->BuildShape())[kM… in InferShape() local
71 if (!x_min_shape.empty() && !x_max_shape.empty()) { in InferShape()
76 max_shape.push_back(x_max_shape[LongToSize(i)]); in InferShape()
|
| D | conv2d.cc | 160 auto x_max_shape = x_shape_map[kMaxShape]; in Conv2dInferShape() local
163 CheckAndConvertUtils::CheckMinMaxShape(x_shape, &x_min_shape, &x_max_shape); in Conv2dInferShape()
168 CheckShapeAllPositive(prim_name + " x_max_shape", x_max_shape); in Conv2dInferShape()
218 …Conv2DPadFunction(&output_hw_max, &pad_list_max, x_max_shape[h_axis], x_max_shape[w_axis], kernel_… in Conv2dInferShape()
229 output_shape_max = {x_max_shape[n_axis], output_hw_max[0], output_hw_max[1], out_channel}; in Conv2dInferShape()
233 output_shape_max = {x_max_shape[n_axis], out_channel, output_hw_max[0], output_hw_max[1]}; in Conv2dInferShape()
|
| D | mat_mul.cc | 59 ShapeVector x_max_shape = x_shape_map[kMaxShape]; in MatMulInferShape() local
62 CheckAndConvertUtils::CheckMinMaxShape(x_shp, &x_min_shape, &x_max_shape); in MatMulInferShape()
89 make_shape(ret_max_shape, x_max_shape, y_max_shape); in MatMulInferShape()
|
| D | batch_matmul.cc | 71 ShapeVector x_max_shape = x_shape_map[kMaxShape]; in BatchMatmulInferShape() local
74 CheckAndConvertUtils::CheckMinMaxShape(x_shp, &x_min_shape, &x_max_shape); in BatchMatmulInferShape()
115 make_shape(ret_max_shape, x_max_shape, y_max_shape); in BatchMatmulInferShape()
|
| /third_party/mindspore/mindspore/core/abstract/ |
| D | prim_nn.cc | 138 ShapeVector x_max_shape = input_x->shape()->max_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()
253 ShapeVector x_max_shape = input_x->shape()->max_shape(); in InferImplConv2D() local
254 CheckMinMaxShape(x_shape, &x_min_shape, &x_max_shape); in InferImplConv2D()
257 CheckShapeAllPositive(op_name + " x_max_shape", x_max_shape); in InferImplConv2D()
316 …Conv2DPadFunction(&output_hw_max, &pad_list_max, x_max_shape[h_axis], x_max_shape[w_axis], kernel_… in InferImplConv2D()
327 output_shape_max = {x_max_shape[n_axis], output_hw_max[0], output_hw_max[1], out_channel}; in InferImplConv2D()
331 output_shape_max = {x_max_shape[n_axis], out_channel, output_hw_max[0], output_hw_max[1]}; in InferImplConv2D()
358 ShapeVector x_max_shape = x->shape()->max_shape(); in InferImplBiasAdd() local
[all …]
|
| D | prim_maths.cc | 288 ShapeVector x_max_shape = x->shape()->max_shape(); in InferImplMatMul() local
291 CheckMinMaxShape(x_shp, &x_min_shape, &x_max_shape); in InferImplMatMul()
316 make_shape(ret_max_shape, x_max_shape, y_max_shape); in InferImplMatMul()
355 ShapeVector x_max_shape = x->shape()->max_shape(); in InferImplBatchMatMul() local
358 CheckMinMaxShape(x_shp, &x_min_shape, &x_max_shape); in InferImplBatchMatMul()
396 make_shape(ret_max_shape, x_max_shape, y_max_shape); in InferImplBatchMatMul()
|
| D | prim_arrays.cc | 758 ShapeVector x_max_shape = x->shape()->max_shape(); in InferImplReshape() local
760 if (x_max_shape.empty()) { in InferImplReshape()
761 x_max_shape = x_shape; in InferImplReshape()
786 for (int64_t value : x_max_shape) { in InferImplReshape()
1095 ShapeVector x_max_shape = x->shape()->max_shape(); in InferImplArgMaxWithValue() local
1096 CheckMinMaxShape(x_shape, &x_min_shape, &x_max_shape); in InferImplArgMaxWithValue()
1099 cal_shape(max_shape, x_max_shape); in InferImplArgMaxWithValue()
|
| /third_party/mindspore/mindspore/ops/operations/ |
| D | array_ops.py | 503 x_max_shape = x['max_shape']
505 x_max_shape = x['shape']
510 max_arr_prod = np.prod(x_max_shape)
|