aboutsummaryrefslogtreecommitdiff
path: root/src/backends/reference/workloads/StridedSlice.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/backends/reference/workloads/StridedSlice.cpp')
-rw-r--r--src/backends/reference/workloads/StridedSlice.cpp4
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