diff options
Diffstat (limited to 'src/armnn/layers/QuantizeLayer.cpp')
-rw-r--r-- | src/armnn/layers/QuantizeLayer.cpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/armnn/layers/QuantizeLayer.cpp b/src/armnn/layers/QuantizeLayer.cpp index e37d6f5300..55f23bf251 100644 --- a/src/armnn/layers/QuantizeLayer.cpp +++ b/src/armnn/layers/QuantizeLayer.cpp @@ -23,7 +23,7 @@ std::unique_ptr<IWorkload> QuantizeLayer::CreateWorkload(const IWorkloadFactory& WorkloadInfo info = PrepInfoAndDesc(descriptor); - return factory.CreateQuantize(descriptor, info); + return factory.CreateWorkload(LayerType::Quantize, descriptor, info); } Layer* QuantizeLayer::Clone(Graph& graph) const |