diff options
Diffstat (limited to 'reference_model/src/ops/ewise_ternary.cc')
| -rw-r--r-- | reference_model/src/ops/ewise_ternary.cc | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/reference_model/src/ops/ewise_ternary.cc b/reference_model/src/ops/ewise_ternary.cc index db5c240..4d53ae4 100644 --- a/reference_model/src/ops/ewise_ternary.cc +++ b/reference_model/src/ops/ewise_ternary.cc @@ -36,6 +36,10 @@ OpSelectBase<Rank, Dtype>::~OpSelectBase() template <int Rank, DType Dtype> int OpSelectBase<Rank, Dtype>::checkTensorAttributes() { + // Check Tosa Level + auto tosa_level = g_func_config.tosa_level; + LEVEL_CHECK(Rank <= tosa_level.MAX_RANK, "Rank should be smaller than or equal to MAX_RANK"); + if (validateRequiredOperands()) return 1; |