Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/infer/
Dsplit_infer.c74 int SplitInferShape(const TensorC *const *inputs, size_t inputs_size, TensorC **outputs, size_t out… in SplitInferShape() function
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/graph/
Dpreprocess_dynamic_shape.cc439 int SplitInferShape(const CNodePtr &cnode, const std::vector<ShapeVector> &in_shapes, in SplitInferShape() function