|
template<typename T , typename std::enable_if< std::is_integral< T >::value, int >::type * > |
SimpleTensor< T > | batch_normalization_layer (const SimpleTensor< T > &src, const SimpleTensor< T > &mean, const SimpleTensor< T > &var, const SimpleTensor< T > &beta, const SimpleTensor< T > &gamma, float epsilon, ActivationLayerInfo act_info, int fixed_point_position) |
|
template SimpleTensor< float > | batch_normalization_layer (const SimpleTensor< float > &src, const SimpleTensor< float > &mean, const SimpleTensor< float > &var, const SimpleTensor< float > &beta, const SimpleTensor< float > &gamma, float epsilon, ActivationLayerInfo act_info, int fixed_point_position) |
|
template SimpleTensor< int8_t > | batch_normalization_layer (const SimpleTensor< int8_t > &src, const SimpleTensor< int8_t > &mean, const SimpleTensor< int8_t > &var, const SimpleTensor< int8_t > &beta, const SimpleTensor< int8_t > &gamma, float epsilon, ActivationLayerInfo act_info, int fixed_point_position) |
|
template SimpleTensor< int16_t > | batch_normalization_layer (const SimpleTensor< int16_t > &src, const SimpleTensor< int16_t > &mean, const SimpleTensor< int16_t > &var, const SimpleTensor< int16_t > &beta, const SimpleTensor< int16_t > &gamma, float epsilon, ActivationLayerInfo act_info, int fixed_point_position) |
|
template SimpleTensor< half > | batch_normalization_layer (const SimpleTensor< half > &src, const SimpleTensor< half > &mean, const SimpleTensor< half > &var, const SimpleTensor< half > &beta, const SimpleTensor< half > &gamma, float epsilon, ActivationLayerInfo act_info, int fixed_point_position) |
|