Searched refs:y_max_shape (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/mindspore/core/ops/ |
| D | mat_mul.cc | 61 ShapeVector y_max_shape = y_shape_map[kMaxShape]; in MatMulInferShape() local
63 CheckAndConvertUtils::CheckMinMaxShape(y_shp, &y_min_shape, &y_max_shape); in MatMulInferShape()
89 make_shape(ret_max_shape, x_max_shape, y_max_shape); in MatMulInferShape()
|
| D | op_utils.cc | 68 auto y_max_shape = y_shape_map[kMaxShape]; in BroadCastInferShape() local
74 auto max_broadcast_shape = CalBroadCastShape(x_max_shape, y_max_shape, op_name); in BroadCastInferShape()
|
| D | batch_matmul.cc | 73 ShapeVector y_max_shape = y_shape_map[kMaxShape]; in BatchMatmulInferShape() local
75 CheckAndConvertUtils::CheckMinMaxShape(y_shp, &y_min_shape, &y_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_maths.cc | 290 ShapeVector y_max_shape = y->shape()->max_shape(); in InferImplMatMul() local
292 CheckMinMaxShape(y_shp, &y_min_shape, &y_max_shape); in InferImplMatMul()
316 make_shape(ret_max_shape, x_max_shape, y_max_shape); in InferImplMatMul()
357 ShapeVector y_max_shape = y->shape()->max_shape(); in InferImplBatchMatMul() local
359 CheckMinMaxShape(y_shp, &y_min_shape, &y_max_shape); in InferImplBatchMatMul()
396 make_shape(ret_max_shape, x_max_shape, y_max_shape); in InferImplBatchMatMul()
|