Home
last modified time | relevance | path

Searched defs:x_min_shape (Results 1 – 9 of 9) sorted by relevance

/third_party/mindspore/mindspore/core/ops/
Dreshape.cc62 ShapeVector x_min_shape = x->shape()->min_shape(); in ReshapeInfer() local
Dop_utils.cc65 auto x_min_shape = x_shape_map[kMinShape]; in BroadCastInferShape() local
Dtranspose.cc32 …auto x_min_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[0]->BuildShape())[kM… in InferShape() local
Dmat_mul.cc58 ShapeVector x_min_shape = x_shape_map[kMinShape]; in MatMulInferShape() local
Dbatch_matmul.cc70 ShapeVector x_min_shape = x_shape_map[kMinShape]; in BatchMatmulInferShape() local
Dconv2d.cc159 auto x_min_shape = x_shape_map[kMinShape]; in Conv2dInferShape() local
/third_party/mindspore/mindspore/core/abstract/
Dprim_nn.cc137 ShapeVector x_min_shape = input_x->shape()->min_shape(); in InferImplBatchNorm() local
252 ShapeVector x_min_shape = input_x->shape()->min_shape(); in InferImplConv2D() local
357 ShapeVector x_min_shape = x->shape()->min_shape(); in InferImplBiasAdd() local
Dprim_maths.cc287 ShapeVector x_min_shape = x->shape()->min_shape(); in InferImplMatMul() local
354 ShapeVector x_min_shape = x->shape()->min_shape(); in InferImplBatchMatMul() local
Dprim_arrays.cc759 ShapeVector x_min_shape = x->shape()->min_shape(); in InferImplReshape() local
1094 ShapeVector x_min_shape = x->shape()->min_shape(); in InferImplArgMaxWithValue() local