diff options
| author | Derek Lamberti <derek.lamberti@arm.com> | 2019-12-10 21:12:59 +0000 |
|---|---|---|
| committer | Francis Murtagh <francis.murtagh@arm.com> | 2019-12-31 10:02:58 +0000 |
| commit | 94a88d2b21d9ca3f42dc3435695be31b5591230b (patch) | |
| tree | edf9381fa095561706353773c155a2c922a50dc8 /src/armnn/layers/MeanLayer.cpp | |
| parent | ff3c426ffd799abd66e4280da559384d86702721 (diff) | |
| download | armnn-94a88d2b21d9ca3f42dc3435695be31b5591230b.tar.gz | |
IVGCVSW-4246 Clean build Layers with -Wextra
Change-Id: I649cd2304fb0040164763d31a12fc77c6c3bed87
Signed-off-by: Derek Lamberti <derek.lamberti@arm.com>
Diffstat (limited to 'src/armnn/layers/MeanLayer.cpp')
| -rw-r--r-- | src/armnn/layers/MeanLayer.cpp | 5 |
1 files changed, 2 insertions, 3 deletions
diff --git a/src/armnn/layers/MeanLayer.cpp b/src/armnn/layers/MeanLayer.cpp index c925a3e18f..30b88fa1b9 100644 --- a/src/armnn/layers/MeanLayer.cpp +++ b/src/armnn/layers/MeanLayer.cpp @@ -19,14 +19,13 @@ MeanLayer::MeanLayer(const armnn::MeanDescriptor& param, const char* name) : LayerWithParameters(1, 1, LayerType::Mean, param, name) {} -std::unique_ptr<IWorkload> MeanLayer::CreateWorkload(const armnn::Graph& graph, - const armnn::IWorkloadFactory& factory) const +std::unique_ptr<IWorkload> MeanLayer::CreateWorkload(const armnn::IWorkloadFactory& factory) const { MeanQueueDescriptor descriptor; descriptor.m_Parameters.m_Axis = m_Param.m_Axis; descriptor.m_Parameters.m_KeepDims = m_Param.m_KeepDims; - return factory.CreateMean(descriptor, PrepInfoAndDesc(descriptor, graph)); + return factory.CreateMean(descriptor, PrepInfoAndDesc(descriptor)); } MeanLayer* MeanLayer::Clone(Graph& graph) const |