aboutsummaryrefslogtreecommitdiff
path: root/src/armnn/layers/StackLayer.cpp
diff options
context:
space:
mode:
authorRob Hughes <robert.hughes@arm.com>2019-08-23 10:11:58 +0100
committerRob Hughes <robert.hughes@arm.com>2019-08-23 09:16:16 +0000
commit91e1d89ad35ad081a4d5995d541791fbefdb3060 (patch)
tree57022df335b5ccd38aa697394f97d12e6bfd3db5 /src/armnn/layers/StackLayer.cpp
parent1cd451cdd818e7df83cf60935abc54345fbc40e6 (diff)
downloadarmnn-91e1d89ad35ad081a4d5995d541791fbefdb3060.tar.gz
Fix Windows build:
* CMake "install" commands require a RUNTIME argument for platforms with DLLs (e.g. Windows). * Replace use of non-standard variable length array with vector * Remove unnecessary #include of unistd.h * Add #ifdefs to dynamic backend code to disable for non-Unix platforms where you can't use dlopen etc. We could implement this properly for Windows later using LoadLibrary etc., but for now erroring is fine. * Add missing #include of <algorithm> Change-Id: Ic8ef5fd599b37bf8772510157b6e479819f6a1eb
Diffstat (limited to 'src/armnn/layers/StackLayer.cpp')
-rw-r--r--src/armnn/layers/StackLayer.cpp4
1 files changed, 2 insertions, 2 deletions
diff --git a/src/armnn/layers/StackLayer.cpp b/src/armnn/layers/StackLayer.cpp
index 7f1dbec461..43e0ac3134 100644
--- a/src/armnn/layers/StackLayer.cpp
+++ b/src/armnn/layers/StackLayer.cpp
@@ -38,7 +38,7 @@ std::vector<TensorShape> StackLayer::InferOutputShapes(const std::vector<TensorS
BOOST_ASSERT(axis <= inputNumDimensions);
- unsigned int dimensionSizes[inputNumDimensions + 1];
+ std::vector<unsigned int> dimensionSizes(inputNumDimensions + 1, 0);
for (unsigned int i = 0; i < axis; ++i)
{
dimensionSizes[i] = inputShape[i];
@@ -51,7 +51,7 @@ std::vector<TensorShape> StackLayer::InferOutputShapes(const std::vector<TensorS
dimensionSizes[i] = inputShape[i-1];
}
- TensorShape targetShape = TensorShape(inputNumDimensions + 1, dimensionSizes);
+ TensorShape targetShape = TensorShape(inputNumDimensions + 1, dimensionSizes.data());
return std::vector<TensorShape>({ targetShape });
}