From ca361233108b5f0c7d21faf2b97576a7a768866c Mon Sep 17 00:00:00 2001 From: Matthew Sloyan Date: Thu, 16 Feb 2023 14:50:22 +0000 Subject: GitHub #714: OnnxParser FullyConnectedLayer inferred shape doesn't match * Added reshape before and after FullyConnected to support dimensions > 2. This is now consistent with the Delegate and TfLiteParser. * Refactored AddFullyConnected method to remove duplicate code. Signed-off-by: Matthew Sloyan Signed-off-by: Francis Murtagh Change-Id: I04dfeb38dbcac096c5fcd9dcb5e3821d38ce6550 --- src/armnnOnnxParser/OnnxParser.hpp | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'src/armnnOnnxParser/OnnxParser.hpp') diff --git a/src/armnnOnnxParser/OnnxParser.hpp b/src/armnnOnnxParser/OnnxParser.hpp index c9f321a5b5..20763c0c85 100644 --- a/src/armnnOnnxParser/OnnxParser.hpp +++ b/src/armnnOnnxParser/OnnxParser.hpp @@ -1,5 +1,5 @@ // -// Copyright © 2017,2022 Arm Ltd and Contributors. All rights reserved. +// Copyright © 2017,2022-2023 Arm Ltd and Contributors. All rights reserved. // SPDX-License-Identifier: MIT // #pragma once @@ -196,7 +196,7 @@ private: static const std::map m_ParserFunctions; /// A mapping of an output slot to each of the input slots it should be connected to - /// The outputSlot is from the layer that creates this tensor as one of its ouputs + /// The outputSlot is from the layer that creates this tensor as one of its outputs /// The inputSlots are from the layers that use this tensor as one of their inputs struct TensorSlots { -- cgit v1.2.1