From 135c95544fda260e8ce622cff7835b886a97663f Mon Sep 17 00:00:00 2001
From: Jerry Ge <jerry.ge@arm.com>
Date: Tue, 23 May 2023 20:59:32 +0000
Subject: Add ERROR_IF to incorrect broadcast shapes

Signed-off-by: Jerry Ge <jerry.ge@arm.com>
Change-Id: I7460ad9eed3ed5c7cec6e855a0303753ed28eb1c
---
 reference_model/src/ops/ewise_ternary.h | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

(limited to 'reference_model/src/ops/ewise_ternary.h')

diff --git a/reference_model/src/ops/ewise_ternary.h b/reference_model/src/ops/ewise_ternary.h
index c6970cb..f24dfbe 100644
--- a/reference_model/src/ops/ewise_ternary.h
+++ b/reference_model/src/ops/ewise_ternary.h
@@ -63,7 +63,7 @@ public:
         : OpSelectBase<Rank, Dtype>(sgt_, attribute_, id_)
     {}
     virtual int eval();
-    int broadcast();
+    int broadcast(std::vector<int>& calculated_shape);
 
     using InEigenType = typename OpSelectBase<Rank, Dtype>::InEigenType;
 };
-- 
cgit v1.2.1