diff options
Diffstat (limited to 'src/armnn/layers/SoftmaxLayer.cpp')
| -rw-r--r-- | src/armnn/layers/SoftmaxLayer.cpp | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/armnn/layers/SoftmaxLayer.cpp b/src/armnn/layers/SoftmaxLayer.cpp index 90891b417a..cb70bbc20d 100644 --- a/src/armnn/layers/SoftmaxLayer.cpp +++ b/src/armnn/layers/SoftmaxLayer.cpp @@ -18,10 +18,10 @@ SoftmaxLayer::SoftmaxLayer(const SoftmaxDescriptor ¶m, const char* name) { } -std::unique_ptr<IWorkload> SoftmaxLayer::CreateWorkload(const Graph& graph, const IWorkloadFactory& factory) const +std::unique_ptr<IWorkload> SoftmaxLayer::CreateWorkload(const IWorkloadFactory& factory) const { SoftmaxQueueDescriptor descriptor; - return factory.CreateSoftmax(descriptor, PrepInfoAndDesc(descriptor, graph)); + return factory.CreateSoftmax(descriptor, PrepInfoAndDesc(descriptor)); } SoftmaxLayer* SoftmaxLayer::Clone(Graph& graph) const |