Searched refs:y_shp (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/mindspore/core/ops/ |
| D | batch_matmul.cc | 38 auto y_shp = y_shape_map[kShape]; in BatchMatmulInferShape() local
40 if (x_shp.size() != y_shp.size() || x_shp.size() < dim_limit) { in BatchMatmulInferShape()
42 … << "or equal to 3, while x size = " << x_shp.size() << ", y size = " << y_shp.size(); in BatchMatmulInferShape()
46 if (x_shp[i] != y_shp[i]) { in BatchMatmulInferShape()
48 << "while x1 is " << x_shp[i] << ", x2 is " << y_shp[i]; in BatchMatmulInferShape()
52 std::vector<int> y_last(y_shp.end() - SizeToInt(offset), y_shp.end()); in BatchMatmulInferShape()
60 std::find(y_shp.begin(), y_shp.end(), -1) == y_shp.end()) { in BatchMatmulInferShape()
65 << ", x2 shape " << y_shp << "(transpose_b=" << transpose_b << "})"; in BatchMatmulInferShape()
75 CheckAndConvertUtils::CheckMinMaxShape(y_shp, &y_min_shape, &y_max_shape); in BatchMatmulInferShape()
81 …std::all_of(y_shp.begin(), y_shp.end(), [](int64_t value) { return value != abstract::Shape::SHP_A… in BatchMatmulInferShape()
[all …]
|
| D | mat_mul.cc | 31 auto y_shp = y_shape_map[kShape]; in MatMulInferShape() local
33 if (x_shp.size() != dim_limit || y_shp.size() != dim_limit) { in MatMulInferShape()
35 … << "or equal to 3, while x size = " << x_shp.size() << ", y size = " << y_shp.size(); in MatMulInferShape()
39 std::vector<int> y_last(y_shp.end() - offset, y_shp.end()); in MatMulInferShape()
47 std::find(y_shp.begin(), y_shp.end(), -1) == y_shp.end()) { in MatMulInferShape()
52 << ", x2 shape " << y_shp << "(transpose_b=" << transpose_b << "})"; in MatMulInferShape()
63 CheckAndConvertUtils::CheckMinMaxShape(y_shp, &y_min_shape, &y_max_shape); in MatMulInferShape()
69 …std::all_of(y_shp.begin(), y_shp.end(), [](int64_t value) { return value != abstract::Shape::SHP_A… in MatMulInferShape()
72 auto y_r = y_shp[(transpose_b ? 1 : 0)]; in MatMulInferShape()
87 make_shape(ret_shape, x_shp, y_shp); in MatMulInferShape()
|
| /third_party/mindspore/mindspore/core/abstract/ |
| D | prim_maths.cc | 278 auto y_shp = y->shape()->shape(); in InferImplMatMul() local
280 if (x_shp.size() != SHAPE_SIZE || y_shp.size() != SHAPE_SIZE) { in InferImplMatMul()
292 CheckMinMaxShape(y_shp, &y_min_shape, &y_max_shape); in InferImplMatMul()
296 …bool y_not_dyn = std::all_of(y_shp.begin(), y_shp.end(), [](int64_t value) { return value != Shape… in InferImplMatMul()
299 auto y_row = y_shp[(transpose_b ? 1 : 0)]; in InferImplMatMul()
314 make_shape(ret_shape, x_shp, y_shp); in InferImplMatMul()
344 auto y_shp = y->shape()->shape(); in InferImplBatchMatMul() local
346 if (x_shp.size() != y_shp.size() || x_shp.size() < minimum_shape) { in InferImplBatchMatMul()
359 CheckMinMaxShape(y_shp, &y_min_shape, &y_max_shape); in InferImplBatchMatMul()
363 …bool y_not_dyn = std::all_of(y_shp.begin(), y_shp.end(), [](int64_t value) { return value != Shape… in InferImplBatchMatMul()
[all …]
|
| /third_party/mindspore/tests/ut/cpp/pre_activate/ascend/buffer_fusion/ |
| D | buffer_fusion_test.cc | 293 std::vector<int64_t> y_shp{768, 768}; in TEST_F() local
295 auto y_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, y_shp); in TEST_F()
|