Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/infer/
Dshape_infer.c20 int ShapeInferShape(const TensorC *const *inputs, size_t inputs_size, TensorC **outputs, size_t out… in ShapeInferShape() function
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/graph_kernel/converter/
Dbasic_op_infer_shape.cc79 BaseShapePtr ShapeInferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &inp… in ShapeInferShape() function
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/graph/
Dpreprocess_dynamic_shape.cc426 int ShapeInferShape(const CNodePtr &cnode, const std::vector<ShapeVector> &in_shapes, in ShapeInferShape() function