|
template<typename T > |
void | magmadnn::math::conv2d (Tensor< T > *x, Tensor< T > *w, Tensor< T > *out) |
|
template void | magmadnn::math::conv2d (Tensor< int > *x, Tensor< int > *w, Tensor< int > *out) |
|
template void | magmadnn::math::conv2d (Tensor< float > *x, Tensor< float > *w, Tensor< float > *out) |
|
template void | magmadnn::math::conv2d (Tensor< double > *x, Tensor< double > *w, Tensor< double > *out) |
|
template<typename T > |
void | magmadnn::math::conv2d_grad_data (Tensor< T > *w, Tensor< T > *grad, Tensor< T > *out) |
|
template void | magmadnn::math::conv2d_grad_data (Tensor< int > *w, Tensor< int > *grad, Tensor< int > *out) |
|
template void | magmadnn::math::conv2d_grad_data (Tensor< float > *w, Tensor< float > *grad, Tensor< float > *out) |
|
template void | magmadnn::math::conv2d_grad_data (Tensor< double > *w, Tensor< double > *grad, Tensor< double > *out) |
|
template<typename T > |
void | magmadnn::math::conv2d_grad_filter (Tensor< T > *w, Tensor< T > *grad, Tensor< T > *out) |
|
template void | magmadnn::math::conv2d_grad_filter (Tensor< int > *x, Tensor< int > *grad, Tensor< int > *out) |
|
template void | magmadnn::math::conv2d_grad_filter (Tensor< float > *x, Tensor< float > *grad, Tensor< float > *out) |
|
template void | magmadnn::math::conv2d_grad_filter (Tensor< double > *x, Tensor< double > *grad, Tensor< double > *out) |
|