diff options
author | Sadik Armagan <sadik.armagan@arm.com> | 2021-02-11 13:57:07 +0000 |
---|---|---|
committer | Sadik Armagan <sadik.armagan@arm.com> | 2021-02-11 13:57:07 +0000 |
commit | 49bdb794170c3d25e3e51fc7b4c267c3d8dbcebf (patch) | |
tree | 1dcfc4495d48320e27354e7a2ff85b6aa3ada5ea /src/armnnSerializer | |
parent | a35b40bb58b2de20893f64a5941ccc54d64a8ffa (diff) | |
download | armnn-49bdb794170c3d25e3e51fc7b4c267c3d8dbcebf.tar.gz |
MLCE-360 'ReduceLayer InferOutputShape Issue'
* Updated ParseReduce() function in TfLiteParser to read correct axis data
* Remove unused m_TargetWidth and m_TargetHight from ReduceDescriptor
* Updated the ArmNN Serializer Schema
Signed-off-by: Sadik Armagan <sadik.armagan@arm.com>
Change-Id: I98b6e00ccba1d8ea8c845cb1ae28840e42339629
Diffstat (limited to 'src/armnnSerializer')
-rw-r--r-- | src/armnnSerializer/ArmnnSchema.fbs | 2 | ||||
-rw-r--r-- | src/armnnSerializer/ArmnnSchema_generated.h | 30 | ||||
-rw-r--r-- | src/armnnSerializer/Serializer.cpp | 2 |
3 files changed, 3 insertions, 31 deletions
diff --git a/src/armnnSerializer/ArmnnSchema.fbs b/src/armnnSerializer/ArmnnSchema.fbs index aa539b188f..9dbf6aa3df 100644 --- a/src/armnnSerializer/ArmnnSchema.fbs +++ b/src/armnnSerializer/ArmnnSchema.fbs @@ -895,8 +895,6 @@ table ReduceLayer { } table ReduceDescriptor { - targetHeight:uint; - targetWidth:uint; keepDims:bool = false; axis:[uint]; reduceOperation:ReduceOperation = Sum; diff --git a/src/armnnSerializer/ArmnnSchema_generated.h b/src/armnnSerializer/ArmnnSchema_generated.h index 32548b2144..cb9e686ca9 100644 --- a/src/armnnSerializer/ArmnnSchema_generated.h +++ b/src/armnnSerializer/ArmnnSchema_generated.h @@ -9207,18 +9207,10 @@ inline flatbuffers::Offset<ReduceLayer> CreateReduceLayer( struct ReduceDescriptor FLATBUFFERS_FINAL_CLASS : private flatbuffers::Table { typedef ReduceDescriptorBuilder Builder; enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE { - VT_TARGETHEIGHT = 4, - VT_TARGETWIDTH = 6, - VT_KEEPDIMS = 8, - VT_AXIS = 10, - VT_REDUCEOPERATION = 12 + VT_KEEPDIMS = 4, + VT_AXIS = 6, + VT_REDUCEOPERATION = 8 }; - uint32_t targetHeight() const { - return GetField<uint32_t>(VT_TARGETHEIGHT, 0); - } - uint32_t targetWidth() const { - return GetField<uint32_t>(VT_TARGETWIDTH, 0); - } bool keepDims() const { return GetField<uint8_t>(VT_KEEPDIMS, 0) != 0; } @@ -9230,8 +9222,6 @@ struct ReduceDescriptor FLATBUFFERS_FINAL_CLASS : private flatbuffers::Table { } bool Verify(flatbuffers::Verifier &verifier) const { return VerifyTableStart(verifier) && - VerifyField<uint32_t>(verifier, VT_TARGETHEIGHT) && - VerifyField<uint32_t>(verifier, VT_TARGETWIDTH) && VerifyField<uint8_t>(verifier, VT_KEEPDIMS) && VerifyOffset(verifier, VT_AXIS) && verifier.VerifyVector(axis()) && @@ -9244,12 +9234,6 @@ struct ReduceDescriptorBuilder { typedef ReduceDescriptor Table; flatbuffers::FlatBufferBuilder &fbb_; flatbuffers::uoffset_t start_; - void add_targetHeight(uint32_t targetHeight) { - fbb_.AddElement<uint32_t>(ReduceDescriptor::VT_TARGETHEIGHT, targetHeight, 0); - } - void add_targetWidth(uint32_t targetWidth) { - fbb_.AddElement<uint32_t>(ReduceDescriptor::VT_TARGETWIDTH, targetWidth, 0); - } void add_keepDims(bool keepDims) { fbb_.AddElement<uint8_t>(ReduceDescriptor::VT_KEEPDIMS, static_cast<uint8_t>(keepDims), 0); } @@ -9273,15 +9257,11 @@ struct ReduceDescriptorBuilder { inline flatbuffers::Offset<ReduceDescriptor> CreateReduceDescriptor( flatbuffers::FlatBufferBuilder &_fbb, - uint32_t targetHeight = 0, - uint32_t targetWidth = 0, bool keepDims = false, flatbuffers::Offset<flatbuffers::Vector<uint32_t>> axis = 0, armnnSerializer::ReduceOperation reduceOperation = armnnSerializer::ReduceOperation_Sum) { ReduceDescriptorBuilder builder_(_fbb); builder_.add_axis(axis); - builder_.add_targetWidth(targetWidth); - builder_.add_targetHeight(targetHeight); builder_.add_reduceOperation(reduceOperation); builder_.add_keepDims(keepDims); return builder_.Finish(); @@ -9289,16 +9269,12 @@ inline flatbuffers::Offset<ReduceDescriptor> CreateReduceDescriptor( inline flatbuffers::Offset<ReduceDescriptor> CreateReduceDescriptorDirect( flatbuffers::FlatBufferBuilder &_fbb, - uint32_t targetHeight = 0, - uint32_t targetWidth = 0, bool keepDims = false, const std::vector<uint32_t> *axis = nullptr, armnnSerializer::ReduceOperation reduceOperation = armnnSerializer::ReduceOperation_Sum) { auto axis__ = axis ? _fbb.CreateVector<uint32_t>(*axis) : 0; return armnnSerializer::CreateReduceDescriptor( _fbb, - targetHeight, - targetWidth, keepDims, axis__, reduceOperation); diff --git a/src/armnnSerializer/Serializer.cpp b/src/armnnSerializer/Serializer.cpp index a2217a3dc4..28afac7b62 100644 --- a/src/armnnSerializer/Serializer.cpp +++ b/src/armnnSerializer/Serializer.cpp @@ -911,8 +911,6 @@ void SerializerVisitor::VisitReduceLayer(const armnn::IConnectableLayer* layer, { auto fbReduceBaseLayer = CreateLayerBase(layer, serializer::LayerType::LayerType_Reduce); auto fbDescriptor = CreateReduceDescriptor(m_flatBufferBuilder, - reduceDescriptor.m_TargetHeight, - reduceDescriptor.m_TargetWidth, reduceDescriptor.m_KeepDims, m_flatBufferBuilder.CreateVector(reduceDescriptor.m_vAxis), GetFlatBufferReduceOperation(reduceDescriptor.m_ReduceOperation)); |