diff options
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(); |