Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/core/ops/
Dop_utils.cc68 auto y_max_shape = y_shape_map[kMaxShape]; in BroadCastInferShape() local
Dmat_mul.cc61 ShapeVector y_max_shape = y_shape_map[kMaxShape]; in MatMulInferShape() local
Dbatch_matmul.cc73 ShapeVector y_max_shape = y_shape_map[kMaxShape]; in BatchMatmulInferShape() local
/third_party/mindspore/mindspore/core/abstract/
Dprim_maths.cc290 ShapeVector y_max_shape = y->shape()->max_shape(); in InferImplMatMul() local
357 ShapeVector y_max_shape = y->shape()->max_shape(); in InferImplBatchMatMul() local