diff options
Diffstat (limited to 'tests/validation/reference/QuantizationLayer.h')
-rw-r--r-- | tests/validation/reference/QuantizationLayer.h | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tests/validation/reference/QuantizationLayer.h b/tests/validation/reference/QuantizationLayer.h index 462396f131..0e80b4906a 100644 --- a/tests/validation/reference/QuantizationLayer.h +++ b/tests/validation/reference/QuantizationLayer.h @@ -35,8 +35,8 @@ namespace validation { namespace reference { -template <typename T> -SimpleTensor<uint8_t> quantization_layer(const SimpleTensor<T> &src, const QuantizationInfo &quantization_info); +template <typename Tin, typename Tout> +SimpleTensor<Tout> quantization_layer(const SimpleTensor<Tin> &src, DataType output_data_type, const QuantizationInfo &quantization_info); } // namespace reference } // namespace validation } // namespace test |