24 #ifndef ARM_COMPUTE_TEST_POOLING_LAYER_H 25 #define ARM_COMPUTE_TEST_POOLING_LAYER_H 38 template <typename T, typename ACC_T = T, typename std::enable_if<is_floating_point<T>::value,
int>
::type = 0>
42 SimpleTensor<T>
pooling_layer(
const SimpleTensor<T> &
src,
const PoolingLayerInfo &
info,
const QuantizationInfo &output_qinfo, SimpleTensor<uint32_t> *indices,
const DataLayout data_layout
decltype(strategy::transforms) typedef type
SimpleTensor< float > src
Copyright (c) 2017-2021 Arm Limited.
Num samples, channels, height, width.
SimpleTensor< T > pooling_layer_internal(const SimpleTensor< T > &src, const PoolingLayerInfo &info, SimpleTensor< uint32_t > *indices, DataLayout data_layout)
ScaleKernelInfo info(interpolation_policy, default_border_mode, PixelValue(), sampling_policy, false)
SimpleTensor< T > pooling_layer(const SimpleTensor< T > &src, const PoolingLayerInfo &info, const QuantizationInfo &output_qinfo, SimpleTensor< uint32_t > *indices, DataLayout data_layout)
DataLayout
[DataLayout enum definition]