#include "tensor/tensor.h"
#include "utilities_internal.h"

Include dependency graph for pooling.h:

This graph shows which files directly or indirectly include this file:

Enumerations
enum	pooling_mode { MAX_POOL, AVERAGE_POOL }

Functions
template<typename T >
void	magmadnn::math::pooling (Tensor< T > x, Tensor< T > out)

template<typename T >
void	magmadnn::math::pooling_grad (Tensor< T > x, Tensor< T > y, Tensor< T > grad, Tensor< T > out)

Detailed Description

Enumerations