Searched refs:make_shape (Results 1 – 3 of 3) sorted by relevance
| /third_party/mindspore/mindspore/core/ops/ |
| D | mat_mul.cc | 81 auto make_shape = [&transpose_a, &transpose_b](ShapeVector &output, const ShapeVector xshp, in MatMulInferShape() local
87 make_shape(ret_shape, x_shp, y_shp); in MatMulInferShape()
88 make_shape(ret_min_shape, x_min_shape, y_min_shape); in MatMulInferShape()
89 make_shape(ret_max_shape, x_max_shape, y_max_shape); in MatMulInferShape()
|
| D | batch_matmul.cc | 95 …auto make_shape = [&transpose_a, &transpose_b, &offset](ShapeVector &output, const ShapeVector xsh… in BatchMatmulInferShape() local
113 make_shape(ret_shape, x_shp, y_shp); in BatchMatmulInferShape()
114 make_shape(ret_min_shape, x_min_shape, y_min_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 | 308 auto make_shape = [&transpose_a, &transpose_b](ShapeVector &output, const ShapeVector xshp, in InferImplMatMul() local
314 make_shape(ret_shape, x_shp, y_shp); in InferImplMatMul()
315 make_shape(ret_min_shape, x_min_shape, y_min_shape); in InferImplMatMul()
316 make_shape(ret_max_shape, x_max_shape, y_max_shape); in InferImplMatMul()
376 auto make_shape = [&transpose_a, &transpose_b](ShapeVector &output, const ShapeVector xshp, in InferImplBatchMatMul() local
394 make_shape(ret_shape, x_shp, y_shp); in InferImplBatchMatMul()
395 make_shape(ret_min_shape, x_min_shape, y_min_shape); in InferImplBatchMatMul()
396 make_shape(ret_max_shape, x_max_shape, y_max_shape); in InferImplBatchMatMul()
|