Home
last modified time | relevance | path

Searched defs:SpaceToDepthInferShape (Results 1 – 2 of 2) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/infer/
Dspace_to_depth_infer.c21 int SpaceToDepthInferShape(const TensorC *const *inputs, size_t inputs_size, TensorC **outputs, siz… in SpaceToDepthInferShape() function
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/
Dspace_to_depth.cc58 abstract::ShapePtr SpaceToDepthInferShape(const PrimitivePtr &primitive, in SpaceToDepthInferShape() function