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<is_floating_point<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);
decltype(strategy::transforms) typedef type
SimpleTensor< float > src
Copyright (c) 2017-2021 Arm Limited.
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)