SHL 2.2.x
|
#include <csinn_data_structure.h>
Public Attributes | |
struct csinn_params_base | base |
int32_t | pool_type |
int32_t | filter_height |
int32_t | filter_width |
int32_t | filter_depth |
int32_t | stride_height |
int32_t | stride_width |
int32_t | stride_depth |
int32_t | pad_top |
int32_t | pad_left |
int32_t | pad_down |
int32_t | pad_right |
int32_t | pad_front |
int32_t | pad_back |
int32_t | ceil_mode |
bool | count_include_pad |
CSI-NN pooling params
struct csinn_params_base csinn_pool_params::base |
The basic information of the operator
int32_t csinn_pool_params::pool_type |
Pool type
int32_t csinn_pool_params::filter_height |
Pool height
int32_t csinn_pool_params::filter_width |
Pool width
int32_t csinn_pool_params::filter_depth |
Pool depth
int32_t csinn_pool_params::stride_height |
Vertical step
int32_t csinn_pool_params::stride_width |
Horizontal step
int32_t csinn_pool_params::stride_depth |
Step in the depth direction
int32_t csinn_pool_params::pad_top |
The number of top padding
int32_t csinn_pool_params::pad_left |
The number of left padding
int32_t csinn_pool_params::pad_down |
The number of bottom padding
int32_t csinn_pool_params::pad_right |
The number of right padding
int32_t csinn_pool_params::pad_front |
The number of front padding
int32_t csinn_pool_params::pad_back |
The number of back padding
int32_t csinn_pool_params::ceil_mode |
When True, will use ceil instead of floor to compute the output shape
bool csinn_pool_params::count_include_pad |
Specifies whether to include the edge pad during pooling, which usually occurs in mean pooling