Home
last modified time | relevance | path

Searched defs:y_shp (Results 1 – 4 of 4) sorted by relevance

/third_party/mindspore/mindspore/core/ops/
Dmat_mul.cc31 auto y_shp = y_shape_map[kShape]; in MatMulInferShape() local
Dbatch_matmul.cc38 auto y_shp = y_shape_map[kShape]; in BatchMatmulInferShape() local
/third_party/mindspore/mindspore/core/abstract/
Dprim_maths.cc278 auto y_shp = y->shape()->shape(); in InferImplMatMul() local
344 auto y_shp = y->shape()->shape(); in InferImplBatchMatMul() local
/third_party/mindspore/tests/ut/cpp/pre_activate/ascend/buffer_fusion/
Dbuffer_fusion_test.cc293 std::vector<int64_t> y_shp{768, 768}; in TEST_F() local