|
template<typename T > |
void | magmadnn::math::pooling (Tensor< T > *x, Tensor< T > *out) |
|
template void | magmadnn::math::pooling (Tensor< int > *x, Tensor< int > *out) |
|
template void | magmadnn::math::pooling (Tensor< float > *x, Tensor< float > *out) |
|
template void | magmadnn::math::pooling (Tensor< double > *x, Tensor< double > *out) |
|
template<typename T > |
void | magmadnn::math::pooling_grad (Tensor< T > *x, Tensor< T > *y, Tensor< T > *grad, Tensor< T > *out) |
|
template void | magmadnn::math::pooling_grad (Tensor< int > *x, Tensor< int > *y, Tensor< int > *grad, Tensor< int > *out) |
|
template void | magmadnn::math::pooling_grad (Tensor< float > *x, Tensor< float > *y, Tensor< float > *grad, Tensor< float > *out) |
|
template void | magmadnn::math::pooling_grad (Tensor< double > *x, Tensor< double > *y, Tensor< double > *grad, Tensor< double > *out) |
|