diff options
Diffstat (limited to 'reference_model/src/ops/activation_funcs.cc')
-rw-r--r-- | reference_model/src/ops/activation_funcs.cc | 12 |
1 files changed, 12 insertions, 0 deletions
diff --git a/reference_model/src/ops/activation_funcs.cc b/reference_model/src/ops/activation_funcs.cc index dc85088..24bd077 100644 --- a/reference_model/src/ops/activation_funcs.cc +++ b/reference_model/src/ops/activation_funcs.cc @@ -26,6 +26,10 @@ using namespace tosa; template <int Rank, DType Dtype> int OpClamp<Rank, Dtype>::register_fcn() { + // 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"); + switch (Dtype) { case DType_FP16: @@ -64,6 +68,10 @@ OpClamp<Rank, Dtype>::~OpClamp() template <int Rank, DType Dtype> int OpSigmoid<Rank, Dtype>::register_fcn() { + // 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"); + switch (Dtype) { case DType_FP16: @@ -83,6 +91,10 @@ int OpSigmoid<Rank, Dtype>::register_fcn() template <int Rank, DType Dtype> int OpTanh<Rank, Dtype>::register_fcn() { + // 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"); + switch (Dtype) { case DType_FP16: |