21.05
|
Pooling Layer Information struct. More...
#include <Types.h>
Public Member Functions | |
PoolingLayerInfo () | |
Default Constructor. More... | |
PoolingLayerInfo (PoolingType pool_type, unsigned int pool_size, DataLayout data_layout, PadStrideInfo pad_stride_info=PadStrideInfo(), bool exclude_padding=false, bool fp_mixed_precision=false) | |
Constructor. More... | |
PoolingLayerInfo (PoolingType pool_type, Size2D pool_size, DataLayout data_layout, PadStrideInfo pad_stride_info=PadStrideInfo(), bool exclude_padding=false, bool fp_mixed_precision=false) | |
Constructor. More... | |
PoolingLayerInfo (PoolingType pool_type, DataLayout data_layout) | |
Constructor. More... | |
Data Fields | |
PoolingType | pool_type |
Size2D | pool_size |
DataLayout | data_layout |
PadStrideInfo | pad_stride_info |
bool | exclude_padding |
bool | is_global_pooling |
bool | fp_mixed_precision |
|
inline |
|
inlineexplicit |
Constructor.
[in] | pool_type | Pooling type PoolingType. |
[in] | pool_size | Pooling size, in elements, across x and y. |
[in] | data_layout | Data layout used by the layer DataLayout |
[in] | pad_stride_info | (Optional) Padding and stride information PadStrideInfo |
[in] | exclude_padding | (Optional) Strategy when accounting padding in calculations. True will exclude padding while false will not (Used in AVG/L2 pooling to determine the pooling area). Defaults to false; |
[in] | fp_mixed_precision | (Optional) Use wider accumulators (32 bit instead of 16 for FP16) to improve accuracy. |
Definition at line 1166 of file Types.h.
|
inlineexplicit |
Constructor.
[in] | pool_type | Pooling type PoolingType. |
[in] | pool_size | Pooling size, in elements, across x and y. |
[in] | data_layout | Data layout used by the layer DataLayout |
[in] | pad_stride_info | (Optional) Padding and stride information PadStrideInfo |
[in] | exclude_padding | (Optional) Strategy when accounting padding in calculations. True will exclude padding while false will not (Used in AVG/L2 pooling to determine the pooling area). Defaults to false; |
[in] | fp_mixed_precision | (Optional) Use wider accumulators (32 bit instead of 16 for FP16) to improve accuracy. |
Definition at line 1193 of file Types.h.
|
inlineexplicit |
Constructor.
[in] | pool_type | Pooling type PoolingType. |
[in] | data_layout | Data layout used by the layer DataLayout |
Definition at line 1216 of file Types.h.
DataLayout data_layout |
Definition at line 1229 of file Types.h.
Referenced by CpuPoolingKernel::configure(), ClPoolingKernel::configure(), CpuPooling::configure(), and CpuPoolingKernel::validate().
bool exclude_padding |
Definition at line 1231 of file Types.h.
Referenced by ClPoolingKernel::configure(), CpuPooling::configure(), arm_compute::cpu::poolingMxN_fp32_neon_nhwc(), and arm_compute::cpu::poolingMxN_q8_neon_nhwc().
bool fp_mixed_precision |
Definition at line 1233 of file Types.h.
Referenced by ClPoolingKernel::configure().
bool is_global_pooling |
Definition at line 1232 of file Types.h.
Referenced by arm_compute::misc::shape_calculator::compute_pool_shape(), CpuPoolingKernel::configure(), ClPoolingKernel::configure(), arm_compute::cpu::poolingMxN_fp32_neon_nhwc(), arm_compute::cpu::poolingMxN_q8_neon_nhwc(), and CpuPoolingKernel::validate().
PadStrideInfo pad_stride_info |
Definition at line 1230 of file Types.h.
Referenced by arm_compute::misc::shape_calculator::compute_pool_shape(), arm_compute::misc::shape_calculator::compute_unpool_shape(), CpuPoolingKernel::configure(), ClPoolingKernel::configure(), arm_compute::cpu::poolingMxN_fp32_neon_nhwc(), arm_compute::cpu::poolingMxN_q8_neon_nhwc(), ClPoolingKernel::run_op(), CpuPoolingKernel::run_op(), CpuPoolingKernel::validate(), and DotGraphVisitor::visit().
Size2D pool_size |
Definition at line 1228 of file Types.h.
Referenced by arm_compute::misc::shape_calculator::compute_pool_shape(), arm_compute::misc::shape_calculator::compute_unpool_shape(), CpuPoolingKernel::configure(), ClPoolingKernel::configure(), CpuPooling::configure(), arm_compute::cpu::poolingMxN_fp32_neon_nhwc(), arm_compute::cpu::poolingMxN_q8_neon_nhwc(), ClPoolingKernel::run_op(), CpuPoolingKernel::run_op(), CpuPoolingKernel::validate(), and DotGraphVisitor::visit().
PoolingType pool_type |
Definition at line 1227 of file Types.h.
Referenced by ClPoolingKernel::configure(), CpuPooling::configure(), arm_compute::cpu::poolingMxN_fp32_neon_nhwc(), arm_compute::cpu::poolingMxN_q8_neon_nhwc(), ClPoolingKernel::run_op(), and DotGraphVisitor::visit().