diff options
Diffstat (limited to 'src/armnnOnnxParser/OnnxParser.hpp')
| -rw-r--r-- | src/armnnOnnxParser/OnnxParser.hpp | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/src/armnnOnnxParser/OnnxParser.hpp b/src/armnnOnnxParser/OnnxParser.hpp index 7716e50fff..f618ff43fd 100644 --- a/src/armnnOnnxParser/OnnxParser.hpp +++ b/src/armnnOnnxParser/OnnxParser.hpp @@ -128,7 +128,9 @@ private: void ResetParser(); void Cleanup(); - std::pair<armnn::ConstTensor, std::unique_ptr<float[]>> CreateConstTensor(const std::string name); + std::pair<armnn::ConstTensor, std::unique_ptr<float[]>> + CreateConstTensor(const std::string name, + armnn::Optional<armnn::PermutationVector&> permutationVector = armnn::EmptyOptional()); template <typename TypeList, typename Location> void ValidateInputs(const onnx::NodeProto& node, |