diff options
Diffstat (limited to 'src/armnn/layers/FloorLayer.cpp')
-rw-r--r-- | src/armnn/layers/FloorLayer.cpp | 5 |
1 files changed, 2 insertions, 3 deletions
diff --git a/src/armnn/layers/FloorLayer.cpp b/src/armnn/layers/FloorLayer.cpp index 34ad7f628e..148543cf62 100644 --- a/src/armnn/layers/FloorLayer.cpp +++ b/src/armnn/layers/FloorLayer.cpp @@ -18,11 +18,10 @@ FloorLayer::FloorLayer(const char* name) { } -std::unique_ptr<IWorkload> FloorLayer::CreateWorkload(const Graph& graph, - const IWorkloadFactory& factory) const +std::unique_ptr<IWorkload> FloorLayer::CreateWorkload(const IWorkloadFactory& factory) const { FloorQueueDescriptor descriptor; - return factory.CreateFloor(descriptor, PrepInfoAndDesc(descriptor, graph)); + return factory.CreateFloor(descriptor, PrepInfoAndDesc(descriptor)); } FloorLayer* FloorLayer::Clone(Graph& graph) const |