diff options
Diffstat (limited to 'nn/common/operations/Split.cpp')
| -rw-r--r-- | nn/common/operations/Split.cpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/nn/common/operations/Split.cpp b/nn/common/operations/Split.cpp index 290e2c8fe..c6276a6bf 100644 --- a/nn/common/operations/Split.cpp +++ b/nn/common/operations/Split.cpp @@ -28,7 +28,7 @@ template <typename Scalar> bool splitGeneric(const Scalar* inputData, const Shape& inputShape, int32_t axis, const std::vector<Scalar*>* outputDataPtrs, const std::vector<Shape>& outputShapes) { - axis = getDimensionIndex(inputShape, axis); + NN_CHECK(handleNegativeAxis(inputShape, &axis)); int outerSize = 1; for (int i = 0; i < axis; ++i) { outerSize *= inputShape.dimensions[i]; |