| 16 | |
| 17 | namespace MNN { |
| 18 | static std::pair<int, int> _resolveShapeRange(const Op* op, int rank) { |
| 19 | int start = 0; |
| 20 | int end = rank; |
| 21 | if (auto param = op->main_as_ShapeParam()) { |
| 22 | if (param->hasStart()) { |
| 23 | start = param->start(); |
| 24 | if (start < 0) { |
| 25 | start += rank; |
| 26 | } |
| 27 | } |
| 28 | if (param->hasEnd()) { |
| 29 | end = param->end(); |
| 30 | if (end < 0) { |
| 31 | end += rank; |
| 32 | } |
| 33 | } |
| 34 | } |
| 35 | start = std::max(0, std::min(start, rank)); |
| 36 | end = std::max(start, std::min(end, rank)); |
| 37 | return std::make_pair(start, end); |
| 38 | } |
| 39 | |
| 40 | class GeometryShape : public GeometryComputer { |
| 41 | public: |