23.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, bool use_inf_as_limit=true, bool use_kernel_indices=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, bool use_inf_as_limit=true, bool use_kernel_indices=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 |
bool | use_inf_as_limit |
bool | use_kernel_indices |
|
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. |
[in] | use_inf_as_limit | (Optional) Use inf to represent the limits of datatypes range, instead of using "lowest" property of the data type. |
[in] | use_kernel_indices | (Optional) Use kernel indices instead of using source indices while computing indices tensor. |
Definition at line 1230 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. |
[in] | use_inf_as_limit | (Optional) Use inf to represent the limits of datatypes range, instead of using "lowest" property of the data type. |
[in] | use_kernel_indices | (Optional) Use kernel indices instead of using source indices while computing indices tensor. |
Definition at line 1263 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 1290 of file Types.h.
DataLayout data_layout |
Definition at line 1305 of file Types.h.
Referenced by ClPool2dKernel::configure(), CpuPool2dKernel::configure(), CpuPool2d::configure(), arm_compute::to_string(), CpuPool2dKernel::validate(), and CpuPool2dAssemblyWrapperKernel::validate().
bool exclude_padding |
Definition at line 1307 of file Types.h.
Referenced by ClPool2dKernel::configure(), CpuPool2dAssemblyWrapperKernel::is_configured(), arm_compute::is_pool_region_entirely_outside_input(), 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 1309 of file Types.h.
Referenced by ClPool2dKernel::configure(), and arm_compute::test::validation::reference::pooling_layer().
bool is_global_pooling |
Definition at line 1308 of file Types.h.
Referenced by PoolingLayerNode::compute_output_descriptor(), arm_compute::misc::shape_calculator::compute_pool_shape(), ClPool2dKernel::configure(), CpuPool2dKernel::configure(), arm_compute::is_pool_region_entirely_outside_input(), arm_compute::test::validation::reference::pooling_layer_internal(), arm_compute::cpu::poolingMxN_fp32_neon_nhwc(), arm_compute::cpu::poolingMxN_fp32_neon_nhwc_kernel_indices(), arm_compute::cpu::poolingMxN_q8_neon_nhwc(), arm_compute::to_string(), and CpuPool2dKernel::validate().
PadStrideInfo pad_stride_info |
Definition at line 1306 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(), ClPool2dKernel::configure(), CpuPool2dKernel::configure(), CpuPool2dAssemblyWrapperKernel::is_configured(), arm_compute::is_pool_region_entirely_outside_input(), arm_compute::test::validation::reference::pooling_layer_internal(), arm_compute::cpu::poolingMxN_fp32_neon_nhwc(), arm_compute::cpu::poolingMxN_fp32_neon_nhwc_kernel_indices(), arm_compute::cpu::poolingMxN_q8_neon_nhwc(), arm_compute::to_string(), CpuPool2dAssemblyWrapperKernel::validate(), and DotGraphVisitor::visit().
Size2D pool_size |
Definition at line 1304 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(), ClPool2dKernel::configure(), CpuPool2dKernel::configure(), CpuPool2d::configure(), CpuPool2dAssemblyWrapperKernel::is_configured(), arm_compute::is_pool_region_entirely_outside_input(), arm_compute::test::validation::reference::pooling_layer_internal(), arm_compute::cpu::poolingMxN_fp32_neon_nhwc(), arm_compute::cpu::poolingMxN_fp32_neon_nhwc_kernel_indices(), arm_compute::cpu::poolingMxN_q8_neon_nhwc(), arm_compute::to_string(), CpuPool2dKernel::validate(), and DotGraphVisitor::visit().
PoolingType pool_type |
Definition at line 1303 of file Types.h.
Referenced by ClPool2dKernel::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().
bool use_inf_as_limit |
Definition at line 1310 of file Types.h.
Referenced by ClPool2dKernel::configure(), arm_compute::test::validation::reference::pooling_layer_internal(), arm_compute::cpu::poolingMxN_fp32_neon_nhwc(), and arm_compute::cpu::poolingMxN_fp32_neon_nhwc_kernel_indices().
bool use_kernel_indices |
Definition at line 1311 of file Types.h.
Referenced by CpuPool2d::configure(), arm_compute::test::validation::reference::pooling_layer_internal(), and arm_compute::cpu::poolingMxN_fp32_neon_nhwc().