21.08
|
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 1163 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 1190 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 1213 of file Types.h.
DataLayout data_layout |
Definition at line 1226 of file Types.h.
Referenced by CpuPool2dKernel::configure(), ClPool2dKernel::configure(), ClPool2d::configure(), CpuPool2d::configure(), arm_compute::to_string(), CpuPool2dKernel::validate(), and CpuPool2dAssemblyWrapperKernel::validate().
bool exclude_padding |
Definition at line 1228 of file Types.h.
Referenced by ClPool2dKernel::configure(), ClPool2d::configure(), CpuPool2d::configure(), CpuPool2dAssemblyWrapperKernel::is_configured(), arm_compute::test::validation::reference::pooling_layer_internal(), arm_compute::cpu::poolingMxN_fp32_neon_nhwc(), arm_compute::cpu::poolingMxN_q8_neon_nhwc(), and CpuPool2dAssemblyWrapperKernel::validate().
bool fp_mixed_precision |
Definition at line 1230 of file Types.h.
Referenced by ClPool2dKernel::configure(), and arm_compute::test::validation::reference::pooling_layer().
bool is_global_pooling |
Definition at line 1229 of file Types.h.
Referenced by PoolingLayerNode::compute_output_descriptor(), arm_compute::misc::shape_calculator::compute_pool_shape(), CpuPool2dKernel::configure(), ClPool2dKernel::configure(), arm_compute::test::validation::reference::pooling_layer_internal(), arm_compute::cpu::poolingMxN_fp32_neon_nhwc(), arm_compute::cpu::poolingMxN_q8_neon_nhwc(), arm_compute::to_string(), and CpuPool2dKernel::validate().
PadStrideInfo pad_stride_info |
Definition at line 1227 of file Types.h.
Referenced by PoolingLayerNode::compute_output_descriptor(), arm_compute::misc::shape_calculator::compute_pool_shape(), arm_compute::misc::shape_calculator::compute_unpool_shape(), CpuPool2dKernel::configure(), ClPool2dKernel::configure(), CpuPool2dAssemblyWrapperKernel::is_configured(), arm_compute::test::validation::reference::pooling_layer_internal(), arm_compute::cpu::poolingMxN_fp32_neon_nhwc(), arm_compute::cpu::poolingMxN_q8_neon_nhwc(), arm_compute::to_string(), CpuPool2dAssemblyWrapperKernel::validate(), and DotGraphVisitor::visit().
Size2D pool_size |
Definition at line 1225 of file Types.h.
Referenced by PoolingLayerNode::compute_output_descriptor(), arm_compute::misc::shape_calculator::compute_pool_shape(), arm_compute::misc::shape_calculator::compute_unpool_shape(), CpuPool2dKernel::configure(), ClPool2dKernel::configure(), CpuPool2d::configure(), CpuPool2dAssemblyWrapperKernel::is_configured(), arm_compute::test::validation::reference::pooling_layer_internal(), arm_compute::cpu::poolingMxN_fp32_neon_nhwc(), arm_compute::cpu::poolingMxN_q8_neon_nhwc(), arm_compute::to_string(), CpuPool2dKernel::validate(), and DotGraphVisitor::visit().
PoolingType pool_type |
Definition at line 1224 of file Types.h.
Referenced by ClPool2dKernel::configure(), ClPool2d::configure(), CpuPool2d::configure(), CpuPool2dAssemblyWrapperKernel::is_configured(), arm_compute::operator<<(), arm_compute::test::validation::reference::pooling_layer_internal(), arm_compute::cpu::poolingMxN_fp32_neon_nhwc(), arm_compute::cpu::poolingMxN_q8_neon_nhwc(), arm_compute::to_string(), CpuPool2dAssemblyWrapperKernel::validate(), and DotGraphVisitor::visit().