Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/infer/
Dconcat_infer.c31 int ConcatInferShape(const TensorC *const *inputs, size_t inputs_size, TensorC **outputs, size_t ou… in ConcatInferShape() function
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/customize/op_proto/
Dconcat_proto.cc343 IMPLEMT_COMMON_INFERFUNC(ConcatInferShape) { in IMPLEMT_COMMON_INFERFUNC() argument
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/graph/
Dpreprocess_dynamic_shape.cc129 int ConcatInferShape(const CNodePtr &cnode, const std::vector<ShapeVector> &in_shapes, in ConcatInferShape() function