1 files changed, 3 insertions, 70 deletions

diff --git a/tests/InferenceModel.hpp b/tests/InferenceModel.hpp
index d2d2ca3c8e..8e463ec326 100644
--- a/tests/InferenceModel.hpp
+++ b/tests/InferenceModel.hpp
@@ -16,6 +16,7 @@
 #endif
 
 #include <HeapProfiling.hpp>
+#include <TensorIOUtils.hpp>
 
 #include <backendsCommon/BackendRegistry.hpp>
 
@@ -302,75 +303,7 @@ public:
 };
 #endif
 
-template<typename TContainer>
-inline armnn::InputTensors MakeInputTensors(
-    const std::vector<armnn::BindingPointInfo>& inputBindings,
-    const std::vector<TContainer>& inputDataContainers)
-{
-    armnn::InputTensors inputTensors;
-
-    const size_t numInputs = inputBindings.size();
-    if (numInputs != inputDataContainers.size())
-    {
-        throw armnn::Exception(boost::str(boost::format("Number of inputs does not match number of "
-            "tensor data containers: %1% != %2%") % numInputs % inputDataContainers.size()));
-    }
-
-    for (size_t i = 0; i < numInputs; i++)
-    {
-        const armnn::BindingPointInfo& inputBinding = inputBindings[i];
-        const TContainer& inputData = inputDataContainers[i];
-
-        boost::apply_visitor([&](auto&& value)
-                             {
-                                 if (value.size() != inputBinding.second.GetNumElements())
-                                 {
-                                    throw armnn::Exception("Input tensor has incorrect size");
-                                 }
-
-                                 armnn::ConstTensor inputTensor(inputBinding.second, value.data());
-                                 inputTensors.push_back(std::make_pair(inputBinding.first, inputTensor));
-                             },
-                             inputData);
-    }
-
-    return inputTensors;
-}
 
-template<typename TContainer>
-inline armnn::OutputTensors MakeOutputTensors(
-    const std::vector<armnn::BindingPointInfo>& outputBindings,
-    std::vector<TContainer>& outputDataContainers)
-{
-    armnn::OutputTensors outputTensors;
-
-    const size_t numOutputs = outputBindings.size();
-    if (numOutputs != outputDataContainers.size())
-    {
-        throw armnn::Exception(boost::str(boost::format("Number of outputs does not match number of "
-            "tensor data containers: %1% != %2%") % numOutputs % outputDataContainers.size()));
-    }
-
-    for (size_t i = 0; i < numOutputs; i++)
-    {
-        const armnn::BindingPointInfo& outputBinding = outputBindings[i];
-        TContainer& outputData = outputDataContainers[i];
-
-        boost::apply_visitor([&](auto&& value)
-                             {
-                                 if (value.size() != outputBinding.second.GetNumElements())
-                                 {
-                                     throw armnn::Exception("Output tensor has incorrect size");
-                                 }
-
-                                 armnn::Tensor outputTensor(outputBinding.second, value.data());
-                                 outputTensors.push_back(std::make_pair(outputBinding.first, outputTensor));
-                             },
-                             outputData);
-    }
-
-    return outputTensors;
-}
 
 template <typename IParser, typename TDataType>
 class InferenceModel
@@ -615,13 +548,13 @@ private:
     template<typename TContainer>
     armnn::InputTensors MakeInputTensors(const std::vector<TContainer>& inputDataContainers)
     {
-        return ::MakeInputTensors(m_InputBindings, inputDataContainers);
+        return armnnUtils::MakeInputTensors(m_InputBindings, inputDataContainers);
     }
 
     template<typename TContainer>
     armnn::OutputTensors MakeOutputTensors(std::vector<TContainer>& outputDataContainers)
     {
-        return ::MakeOutputTensors(m_OutputBindings, outputDataContainers);
+        return armnnUtils::MakeOutputTensors(m_OutputBindings, outputDataContainers);
     }
 
     std::chrono::high_resolution_clock::time_point GetCurrentTime()