1 files changed, 5 insertions, 3 deletions
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<hal_1_2::HalPolicy>(operation, 0, *layer, model, data);
 }