Home
last modified time | relevance | path

Searched defs:shape_backbone (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/core/ops/ops_func_impl/
Dmatmul_ext.cc84 ShapeVector InferShapeRem(const ShapeVector &shape_backbone, const ShapeVector &shape1, const Shape… in InferShapeRem()
156 ShapeVector shape_backbone = CheckMatMulShapes(x_shp, y_shp); in InferShape() local
205 ShapeVector shape_backbone = CheckMatMulShapes(x_shp, y_shp); in InferShape() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/pyboost/customize/
Dmatmul_ext.cc109 ShapeVector shape_backbone = ops::CheckMatMulShapes(shape1_orig, shape2_orig); in MatMulExtCPUCustomize() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/pyboost/customize/
Dmatmul_ext.cc108 ShapeVector shape_backbone = ops::CheckMatMulShapes(shape1_orig, shape2_orig); in MatMulExtGPUCustomize() local