IVGCVSW-6641 Stabilize the IWorkloadFactory interface with unified strategy
Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com>
Change-Id: Ia941be9bf2c15fe56e49a9b9a2bbe943a8152438
diff --git a/src/armnn/layers/ElementwiseUnaryLayer.cpp b/src/armnn/layers/ElementwiseUnaryLayer.cpp
index 37d6084..c50910b 100644
--- a/src/armnn/layers/ElementwiseUnaryLayer.cpp
+++ b/src/armnn/layers/ElementwiseUnaryLayer.cpp
@@ -23,13 +23,7 @@
std::unique_ptr<IWorkload> ElementwiseUnaryLayer::CreateWorkload(const IWorkloadFactory& factory) const
{
ElementwiseUnaryQueueDescriptor descriptor;
-
- if (descriptor.m_Parameters.m_Operation == UnaryOperation::LogicalNot)
- {
- return factory.CreateLogicalUnary(descriptor, PrepInfoAndDesc(descriptor));
- }
-
- return factory.CreateElementwiseUnary(descriptor, PrepInfoAndDesc(descriptor));
+ return factory.CreateWorkload(LayerType::ElementwiseUnary, descriptor, PrepInfoAndDesc(descriptor));
}
ElementwiseUnaryLayer* ElementwiseUnaryLayer::Clone(Graph& graph) const