24.04
|
Go to the documentation of this file.
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,
@ NCHW
Num samples, channels, height, width.
SimpleTensor< T > pooling_layer_internal(const SimpleTensor< T > &src, const PoolingLayerInfo &info, SimpleTensor< uint32_t > *indices, DataLayout data_layout)
SimpleTensor< float > src
decltype(strategy::transforms) typedef type
DataLayout
[DataLayout enum definition]
SimpleTensor< T > pooling_layer(const SimpleTensor< T > &src, const PoolingLayerInfo &info, const QuantizationInfo &output_qinfo, SimpleTensor< uint32_t > *indices, DataLayout data_layout)
const DataLayout data_layout
Copyright (c) 2017-2024 Arm Limited.
ScaleKernelInfo info(interpolation_policy, default_border_mode, PixelValue(), sampling_policy, false)