Searched defs:ReduceInferShape (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ops_func_impl/ |
| D | reduce_arithmetic.cc | 35 BaseShapePtr ReduceInferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &in… in ReduceInferShape() function
215 ShapeArray ReduceInferShape(const PrimitivePtr &primitive, const ValuePtrList &input_values) { in ReduceInferShape() function
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/infer/ |
| D | reduce_infer.c | 64 int ReduceInferShape(const TensorC *const *inputs, size_t inputs_size, TensorC **outputs, size_t ou… in ReduceInferShape() function
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp32/ |
| D | reduce_fp32.c | 216 int ReduceInferShape(int32_t **in_shape, size_t *dim_size, int32_t *out_shape, int32_t *in_format, … in ReduceInferShape() function
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/graph/ |
| D | preprocess_dynamic_shape.cc | 298 int ReduceInferShape(const CNodePtr &cnode, const std::vector<ShapeVector> &in_shapes, in ReduceInferShape() function
|