24 #ifndef __ARM_COMPUTE_TEST_BATCH_NORMALIZATION_LAYER_H__ 25 #define __ARM_COMPUTE_TEST_BATCH_NORMALIZATION_LAYER_H__ 38 template <typename T, typename std::enable_if<std::is_integral<T>::value,
int>::type * =
nullptr>
39 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,
40 ActivationLayerInfo act_info,
41 int fixed_point_position);
43 template <typename T, typename std::enable_if<is_floating_point<T>::value,
int>::type * =
nullptr>
44 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,
45 ActivationLayerInfo act_info,
46 int fixed_point_position);
This file contains all available output stages for GEMMLowp on OpenCL.
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)
convolution configure & src