diff options
Diffstat (limited to 'reference_model/src/ops/activation_funcs.h')
-rw-r--r-- | reference_model/src/ops/activation_funcs.h | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/reference_model/src/ops/activation_funcs.h b/reference_model/src/ops/activation_funcs.h index 9a697cd..2372fcb 100644 --- a/reference_model/src/ops/activation_funcs.h +++ b/reference_model/src/ops/activation_funcs.h @@ -1,5 +1,5 @@ -// Copyright (c) 2020-2022, ARM Limited. +// Copyright (c) 2020-2023, ARM Limited. // // Licensed under the Apache License, Version 2.0 (the "License"); // you may not use this file except in compliance with the License. @@ -24,7 +24,7 @@ using namespace tosa; namespace TosaReference { -template <int Rank, DType Dtype> +template <int Rank, TOSA_REF_TYPE Dtype> class OpClamp : public UnaryNode<Rank, Dtype> { public: @@ -45,7 +45,7 @@ protected: TosaClampAttribute* attribute; }; -template <int Rank, DType Dtype> +template <int Rank, TOSA_REF_TYPE Dtype> class OpSigmoid : public UnaryNode<Rank, Dtype> { public: @@ -61,7 +61,7 @@ public: virtual int register_fcn(); }; -template <int Rank, DType Dtype> +template <int Rank, TOSA_REF_TYPE Dtype> class OpTanh : public UnaryNode<Rank, Dtype> { public: |