diff options
author | Jerry Ge <jerry.ge@arm.com> | 2023-05-23 20:59:32 +0000 |
---|---|---|
committer | Dominic Symes <dominic.symes@arm.com> | 2023-06-15 18:25:54 +0000 |
commit | 135c95544fda260e8ce622cff7835b886a97663f (patch) | |
tree | 5d46f8f48978112abff037309a827b5844ee80de /reference_model/src/ops/ewise_ternary.h | |
parent | cb7201e173961760c042cade591afe763c949c8f (diff) | |
download | reference_model-135c95544fda260e8ce622cff7835b886a97663f.tar.gz |
Add ERROR_IF to incorrect broadcast shapes
Signed-off-by: Jerry Ge <jerry.ge@arm.com>
Change-Id: I7460ad9eed3ed5c7cec6e855a0303753ed28eb1c
Diffstat (limited to 'reference_model/src/ops/ewise_ternary.h')
-rw-r--r-- | reference_model/src/ops/ewise_ternary.h | 2 |
1 files changed, 1 insertions, 1 deletions
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; }; |