diff options
Diffstat (limited to 'src/armnn/layers/PreluLayer.cpp')
-rw-r--r-- | src/armnn/layers/PreluLayer.cpp | 5 |
1 files changed, 2 insertions, 3 deletions
diff --git a/src/armnn/layers/PreluLayer.cpp b/src/armnn/layers/PreluLayer.cpp index 6040248391..d9e59224a0 100644 --- a/src/armnn/layers/PreluLayer.cpp +++ b/src/armnn/layers/PreluLayer.cpp @@ -18,12 +18,11 @@ PreluLayer::PreluLayer(const char* name) : Layer(2, 1, LayerType::Prelu, name) {} -std::unique_ptr<IWorkload> PreluLayer::CreateWorkload(const Graph& graph, - const IWorkloadFactory& factory) const +std::unique_ptr<IWorkload> PreluLayer::CreateWorkload(const IWorkloadFactory& factory) const { PreluQueueDescriptor descriptor; - return factory.CreatePrelu(descriptor, PrepInfoAndDesc(descriptor, graph)); + return factory.CreatePrelu(descriptor, PrepInfoAndDesc(descriptor)); } PreluLayer* PreluLayer::Clone(Graph& graph) const |