diff options
Diffstat (limited to 'reference_model/src/ops/type_conversion.cc')
| -rw-r--r-- | reference_model/src/ops/type_conversion.cc | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/reference_model/src/ops/type_conversion.cc b/reference_model/src/ops/type_conversion.cc index a3140c3..9034add 100644 --- a/reference_model/src/ops/type_conversion.cc +++ b/reference_model/src/ops/type_conversion.cc @@ -45,6 +45,10 @@ OpRescale<Rank, InDtype, OutDtype>::~OpRescale() template <int Rank, DType InDtype, DType OutDtype> int OpRescale<Rank, InDtype, OutDtype>::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; @@ -250,6 +254,10 @@ OpCast<Rank, InDtype, OutDtype>::~OpCast() template <int Rank, DType InDtype, DType OutDtype> int OpCast<Rank, InDtype, OutDtype>::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; |