diff options
Diffstat (limited to 'src/backends/reference/workloads/StridedSlice.cpp')
-rw-r--r-- | src/backends/reference/workloads/StridedSlice.cpp | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/backends/reference/workloads/StridedSlice.cpp b/src/backends/reference/workloads/StridedSlice.cpp index 9f2b1e76f6..62f06dc5ec 100644 --- a/src/backends/reference/workloads/StridedSlice.cpp +++ b/src/backends/reference/workloads/StridedSlice.cpp @@ -68,7 +68,7 @@ TensorShape ExtendShape(const TensorShape& inputShape, return inputShape; } - unsigned int newSizes[newNumDimensions]; + std::vector<unsigned int> newSizes(newNumDimensions, 0); unsigned int diff = newNumDimensions - inputShape.GetNumDimensions(); @@ -82,7 +82,7 @@ TensorShape ExtendShape(const TensorShape& inputShape, newSizes[i] = inputShape[i - diff]; } - return TensorShape(newNumDimensions, newSizes); + return TensorShape(newNumDimensions, newSizes.data()); } } // Anonymous namespace |