From 0c95f4cd319874ffa4aba3a378e0e3346f688fdc Mon Sep 17 00:00:00 2001 From: Narumol Prangnawarat Date: Wed, 18 Nov 2020 16:52:07 +0000 Subject: Fix logical vts skip * Add Boolean support for Reshape * Use LogicalUnary factory and data type for LogicalNot Signed-off-by: Narumol Prangnawarat Change-Id: I8e072fde200b7716556ae67f79616458cf98ff20 --- src/armnn/layers/ElementwiseUnaryLayer.cpp | 5 +++++ 1 file changed, 5 insertions(+) (limited to 'src/armnn') diff --git a/src/armnn/layers/ElementwiseUnaryLayer.cpp b/src/armnn/layers/ElementwiseUnaryLayer.cpp index 74fa16e15f..8c94106818 100644 --- a/src/armnn/layers/ElementwiseUnaryLayer.cpp +++ b/src/armnn/layers/ElementwiseUnaryLayer.cpp @@ -24,6 +24,11 @@ std::unique_ptr ElementwiseUnaryLayer::CreateWorkload(const IWorkload { ElementwiseUnaryQueueDescriptor descriptor; + if (descriptor.m_Parameters.m_Operation == UnaryOperation::LogicalNot) + { + return factory.CreateLogicalUnary(descriptor, PrepInfoAndDesc(descriptor)); + } + return factory.CreateElementwiseUnary(descriptor, PrepInfoAndDesc(descriptor)); } -- cgit v1.2.1