diff options
Diffstat (limited to 'src/backends/backendsCommon/test/ComparisonEndToEndTestImpl.hpp')
-rw-r--r-- | src/backends/backendsCommon/test/ComparisonEndToEndTestImpl.hpp | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/src/backends/backendsCommon/test/ComparisonEndToEndTestImpl.hpp b/src/backends/backendsCommon/test/ComparisonEndToEndTestImpl.hpp index dc53b7b246..c705f87b85 100644 --- a/src/backends/backendsCommon/test/ComparisonEndToEndTestImpl.hpp +++ b/src/backends/backendsCommon/test/ComparisonEndToEndTestImpl.hpp @@ -10,6 +10,8 @@ #include <armnn/INetwork.hpp> +#include <armnn/utility/NumericCast.hpp> + #include <boost/test/unit_test.hpp> #include <vector> @@ -34,7 +36,7 @@ INetworkPtr CreateComparisonNetwork(const std::vector<TensorShape>& inputShapes, for (unsigned int i = 0; i < inputShapes.size(); ++i) { TensorInfo inputTensorInfo(inputShapes[i], ArmnnTypeInput, qScale, qOffset); - IConnectableLayer* input = net->AddInputLayer(boost::numeric_cast<LayerBindingId>(i)); + IConnectableLayer* input = net->AddInputLayer(armnn::numeric_cast<LayerBindingId>(i)); Connect(input, comparisonLayer, inputTensorInfo, 0, i); } |