From 452274c86245082ce20563ede12b92af81dba38a Mon Sep 17 00:00:00 2001
From: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
Date: Thu, 23 Sep 2021 16:12:19 +0100
Subject: IVGCVSW-6459 Add support of scalar and flexible output datatypes to
 ONNX parser

Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
Change-Id: Id1e933f6ae55ddc1a57c80c9f6a5757ccb61f018
---
 src/armnnOnnxParser/OnnxParser.hpp | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

(limited to 'src/armnnOnnxParser/OnnxParser.hpp')

diff --git a/src/armnnOnnxParser/OnnxParser.hpp b/src/armnnOnnxParser/OnnxParser.hpp
index 6a0fad0ec2..d388f501d4 100644
--- a/src/armnnOnnxParser/OnnxParser.hpp
+++ b/src/armnnOnnxParser/OnnxParser.hpp
@@ -74,9 +74,11 @@ private:
 
     void SetupInfo(const google::protobuf::RepeatedPtrField<onnx::ValueInfoProto >* list);
 
-    std::vector<armnn::TensorInfo> ComputeOutputInfo(std::vector<std::string> outNames,
-                                                     const armnn::IConnectableLayer* layer,
-                                                     std::vector<armnn::TensorShape> inputShapes);
+    std::vector<armnn::TensorInfo> ComputeOutputInfo(
+        std::vector<std::string> outNames,
+        const armnn::IConnectableLayer* layer,
+        std::vector<armnn::TensorShape> inputShapes,
+        const onnx::TensorProto::DataType& type = onnx::TensorProto::FLOAT);
 
     void DetectFullyConnected();
 
-- 
cgit v1.2.1