Searched refs:InferImplReduceFuncCalShape (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/mindspore/core/ops/ |
| D | reduce_sum.cc | 38 void InferImplReduceFuncCalShape(ShapeVector *shape, const ShapeVector &x_shape, const ValuePtr &ax… in InferImplReduceFuncCalShape() function
166 InferImplReduceFuncCalShape(&out_shape, input_shape, axis_value, keep_dims); in InferShape()
171 InferImplReduceFuncCalShape(&shape_min, input_min_shape, axis_value, keep_dims); in InferShape()
172 InferImplReduceFuncCalShape(&shape_max, input_max_shape, axis_value, keep_dims); in InferShape()
|
| /third_party/mindspore/mindspore/core/abstract/ |
| D | prim_maths.cc | 80 void InferImplReduceFuncCalShape(ShapeVector *shape, const ShapeVector &x_shape, const ValuePtr &ax… in InferImplReduceFuncCalShape() function
147 InferImplReduceFuncCalShape(&shape, x_shape, axis, keep_dims_value); in InferImplReduceFunc()
155 InferImplReduceFuncCalShape(&shape_min, x_shape_min, axis, keep_dims_value); in InferImplReduceFunc()
156 InferImplReduceFuncCalShape(&shape_max, x_shape_max, axis, keep_dims_value); in InferImplReduceFunc()
|