diff options
| author | Kevin Cheng <kevin.cheng@arm.com> | 2021-11-02 01:15:50 +0000 |
|---|---|---|
| committer | Kevin Cheng <kevin.cheng@arm.com> | 2021-11-02 01:17:45 +0000 |
| commit | f3e016f7aedad5bc995932e75821f2eeb2ec45ea (patch) | |
| tree | 3f7281724627ca4cc7e4ca1385b74bab4635e27a /reference_model/src/ops/ewise_binary.cc | |
| parent | 80794804907bfe73367f40616f3b6c41deacaca3 (diff) | |
| download | reference_model-f3e016f7aedad5bc995932e75821f2eeb2ec45ea.tar.gz | |
more ERROR_IF fixes
- TRANSPOSE: move perm attribute check to compile-time checker
- TABLE: add output type checker
Signed-off-by: Kevin Cheng <kevin.cheng@arm.com>
Change-Id: I834a5f290fbc384ef339b624060e6e5c77072c36
Diffstat (limited to 'reference_model/src/ops/ewise_binary.cc')
| -rw-r--r-- | reference_model/src/ops/ewise_binary.cc | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/reference_model/src/ops/ewise_binary.cc b/reference_model/src/ops/ewise_binary.cc index 415cd1c..b199f69 100644 --- a/reference_model/src/ops/ewise_binary.cc +++ b/reference_model/src/ops/ewise_binary.cc @@ -512,6 +512,7 @@ int OpTable<Rank, InDtype>::checkTensorAttributes() } ERROR_IF(inputs[0]->getDtype() != InDtype, "OpTable: Unexpected input type"); + ERROR_IF(outputs[0]->getDtype() != OutDtype, "OpTable: Unexpected output type"); ERROR_IF(attribute->table().size() != TableNumEntries, "OpTable: table attribute size must be %u", TableNumEntries); for (uint32_t i = 0; i < TableNumEntries; i++) |