aboutsummaryrefslogtreecommitdiff
path: root/Utils.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'Utils.hpp')
-rw-r--r--Utils.hpp3
1 files changed, 2 insertions, 1 deletions
diff --git a/Utils.hpp b/Utils.hpp
index c4d89f7e..f68747b0 100644
--- a/Utils.hpp
+++ b/Utils.hpp
@@ -9,6 +9,7 @@
#include <CpuExecutor.h>
#include <HalInterfaces.h>
#include <NeuralNetworks.h>
+#include <Utils.h>
#include <vector>
#include <string>
@@ -63,7 +64,7 @@ void SwizzleAndroidNn4dTensorToArmNn(const armnn::TensorInfo& tensor, const void
const armnn::PermutationVector& mappings);
/// Returns a pointer to a specific location in a pool
-void* GetMemoryFromPool(DataLocation location,
+void* GetMemoryFromPool(V1_0::DataLocation location,
const std::vector<android::nn::RunTimePoolInfo>& memPools);
/// Can throw UnsupportedOperand