Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore-src/source/mindspore/core/ops/
Dreservoir_replay_buffer.cc235 BaseShapePtr CommonInferShape(const PrimitivePtr &, const std::vector<AbstractBasePtr> &) { in CommonInferShape() function
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/infer/
Dcommon_infer.c170 int CommonInferShape(const TensorC *const *inputs, size_t inputs_size, TensorC **outputs, size_t ou… in CommonInferShape() function
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/graph/
Dpreprocess_dynamic_shape.cc122 int CommonInferShape(const CNodePtr &cnode, const std::vector<ShapeVector> &in_shapes, in CommonInferShape() function