From deb3bdbe028a59da0759dd7a560387d03a11d322 Mon Sep 17 00:00:00 2001
From: surmeh01 <surabhi.mehta@arm.com>
Date: Thu, 5 Jul 2018 12:06:04 +0100
Subject: Release 18.05.02

---
 Utils.hpp | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

(limited to 'Utils.hpp')

diff --git a/Utils.hpp b/Utils.hpp
index e6b56be0..4b5066ee 100644
--- a/Utils.hpp
+++ b/Utils.hpp
@@ -10,6 +10,8 @@
 #include <armnn/ArmNN.hpp>
 #include <CpuExecutor.h>
 
+#include "ArmnnDriver.hpp"
+
 #include <vector>
 #include <string>
 
@@ -41,7 +43,7 @@ void* GetMemoryFromPool(DataLocation location,
 armnn::TensorInfo GetTensorInfoForOperand(const Operand& operand);
 
 std::string GetOperandSummary(const Operand& operand);
-std::string GetModelSummary(const Model& model);
+std::string GetModelSummary(const V1_0::Model& model);
 
 void DumpTensor(const std::string& dumpDir,
     const std::string& requestName,
@@ -50,5 +52,5 @@ void DumpTensor(const std::string& dumpDir,
 
 void ExportNetworkGraphToDotFile(const armnn::IOptimizedNetwork& optimizedNetwork,
                                  const std::string& dumpDir,
-                                 const Model& model);
+                                 const V1_0::Model& model);
 }
-- 
cgit v1.2.1