Public Member Functions | |
PoolingLayer (op::Operation< T > *input, const std::vector< unsigned int > &filter_shape={2, 2}, const std::vector< unsigned int > &padding={0, 0}, const std::vector< unsigned int > &strides={1, 1}, pooling_mode mode=MAX_POOL, bool propagate_nan=false) | |
virtual std::vector< op::Operation< T > * > | get_weights () |
![]() | |
virtual op::Operation< T > * | out () |
op::Operation< T > * | get_input () |
op::Operation< T > * | get_output () |
std::vector< unsigned int > | get_input_shape () const |
std::vector< unsigned int > | get_output_shape () const |
unsigned int | get_input_shape (unsigned int i) const |
unsigned int | get_output_shape (unsigned int i) const |
void | set_name (std::string name) |
std::string | get_name () const |
Protected Member Functions | |
void | init () |
![]() | |
Layer (std::vector< unsigned int > input_shape, op::Operation< T > *input) | |
Protected Attributes | |
pooling_mode | mode |
bool | propagate_nan |
int | filter_h |
int | filter_w |
int | pad_h |
int | pad_w |
int | stride_h |
int | stride_w |
![]() | |
std::vector< unsigned int > | input_shape |
std::vector< unsigned int > | output_shape |
op::Operation< T > * | input |
op::Operation< T > * | output |
std::string | name |