diff options
author | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2021-09-23 16:12:19 +0100 |
---|---|---|
committer | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2021-10-07 14:43:09 +0000 |
commit | 452274c86245082ce20563ede12b92af81dba38a (patch) | |
tree | 79718c6cf86acbb21138068c17aae15c4b172306 /src/armnnOnnxParser/OnnxParser.hpp | |
parent | 4d217c02fe2c0a32ff9da69d8fe375a75173c0f3 (diff) | |
download | armnn-452274c86245082ce20563ede12b92af81dba38a.tar.gz |
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
Diffstat (limited to 'src/armnnOnnxParser/OnnxParser.hpp')
-rw-r--r-- | src/armnnOnnxParser/OnnxParser.hpp | 8 |
1 files changed, 5 insertions, 3 deletions
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(); |