#include "QuantizationLayer.h"
#include <cmath>
Go to the source code of this file.
|
template<typename Tin , typename Tout > |
SimpleTensor< Tout > | quantization_layer (const SimpleTensor< Tin > &src, DataType output_data_type, const QuantizationInfo &quantization_info) |
|
template<> |
SimpleTensor< uint8_t > | quantization_layer (const SimpleTensor< uint8_t > &src, DataType output_data_type, const QuantizationInfo &quantization_info) |
|
template<> |
SimpleTensor< uint8_t > | quantization_layer (const SimpleTensor< int8_t > &src, DataType output_data_type, const QuantizationInfo &quantization_info) |
|
template SimpleTensor< int8_t > | quantization_layer (const SimpleTensor< half > &src, DataType output_data_type, const QuantizationInfo &quantization_info) |
|
template SimpleTensor< int8_t > | quantization_layer (const SimpleTensor< float > &src, DataType output_data_type, const QuantizationInfo &quantization_info) |
|