From 793a70c1a5b8021137705d312916ac14ca35509a Mon Sep 17 00:00:00 2001 From: Sadik Armagan Date: Thu, 19 Mar 2020 13:54:04 +0000 Subject: IVGCVSW-4565 TENSOR_BOOL8 data type not supported in AndroidNN Driver * Added TENSOR_BOOL8 support * Added Broadcast support to comparision operators !armnn:2903 Signed-off-by: Sadik Armagan Change-Id: I844e32b57399eff2dc60af9b2099145316c80cae --- 1.2/HalPolicy.cpp | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) (limited to '1.2') diff --git a/1.2/HalPolicy.cpp b/1.2/HalPolicy.cpp index b3ccc47f..1811688f 100644 --- a/1.2/HalPolicy.cpp +++ b/1.2/HalPolicy.cpp @@ -279,9 +279,11 @@ bool HalPolicy::ConvertComparison(const Operation& operation, IConnectableLayer* layer = data.m_Network->AddComparisonLayer(descriptor); assert(layer != nullptr); - - input0.Connect(layer->GetInputSlot(0)); - input1.Connect(layer->GetInputSlot(1)); + bool isReshapeSupported = BroadcastTensor(input0, input1, layer, data); + if (!isReshapeSupported) + { + return false; + } return SetupAndTrackLayerOutputSlot(operation, 0, *layer, model, data); } -- cgit v1.2.1