diff options
Diffstat (limited to 'src/backends/reference/workloads')
-rw-r--r-- | src/backends/reference/workloads/Debug.cpp | 2 | ||||
-rw-r--r-- | src/backends/reference/workloads/StridedSlice.cpp | 4 |
2 files changed, 3 insertions, 3 deletions
diff --git a/src/backends/reference/workloads/Debug.cpp b/src/backends/reference/workloads/Debug.cpp index d1c9fdd8b8..09a0dfc03b 100644 --- a/src/backends/reference/workloads/Debug.cpp +++ b/src/backends/reference/workloads/Debug.cpp @@ -23,7 +23,7 @@ void Debug(const TensorInfo& inputInfo, const unsigned int numElements = inputInfo.GetNumElements(); const TensorShape& inputShape = inputInfo.GetShape(); - unsigned int strides[numDims]; + std::vector<unsigned int> strides(numDims, 0); strides[numDims - 1] = inputShape[numDims - 1]; for (unsigned int i = 2; i <= numDims; i++) 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 |