diff options
Diffstat (limited to 'src/armnn/layers/DequantizeLayer.cpp')
-rw-r--r-- | src/armnn/layers/DequantizeLayer.cpp | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/armnn/layers/DequantizeLayer.cpp b/src/armnn/layers/DequantizeLayer.cpp index 4dd30de77b..00a1d697b6 100644 --- a/src/armnn/layers/DequantizeLayer.cpp +++ b/src/armnn/layers/DequantizeLayer.cpp @@ -16,12 +16,12 @@ DequantizeLayer::DequantizeLayer(const char* name) : Layer(1, 1, LayerType::Dequantize, name) {} -std::unique_ptr<IWorkload> DequantizeLayer::CreateWorkload(const Graph& graph, +std::unique_ptr<IWorkload> DequantizeLayer::CreateWorkload( const IWorkloadFactory& factory) const { DequantizeQueueDescriptor descriptor; - return factory.CreateDequantize(descriptor, PrepInfoAndDesc(descriptor, graph)); + return factory.CreateDequantize(descriptor, PrepInfoAndDesc(descriptor)); } DequantizeLayer* DequantizeLayer::Clone(Graph& graph) const |