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_binary.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_binary.h')
| -rw-r--r-- | reference_model/src/ops/ewise_binary.h | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/reference_model/src/ops/ewise_binary.h b/reference_model/src/ops/ewise_binary.h index 5f6e531..3a6f24c 100644 --- a/reference_model/src/ops/ewise_binary.h +++ b/reference_model/src/ops/ewise_binary.h @@ -55,7 +55,7 @@ public: using TOut = Eigen::Tensor<OutEigenType, Rank>; protected: - int broadcast(); + int broadcast(std::vector<int>& calculated_shape); protected: std::function<OutEigenType(InEigenType, InEigenType)> fcn; |