Go to the source code of this file.
|
| template<typename T > |
| using | lbann::ToReal = typename ToRealT< T >::type |
| |
| template<typename T > |
| using | lbann::ToComplex = typename ToComplexT< T >::type |
| |
|
| template<typename T > |
| auto | lbann::fft::get_r2c_output_dims (std::vector< T > const &dims) |
| |
| template<typename InT , typename OutT , typename TransformT > |
| void | lbann::fft::r2c_to_full_1d (El::Matrix< InT, El::Device::CPU > const &r2c_input, El::Matrix< OutT, El::Device::CPU > &full_output, std::vector< int > const &full_dims, TransformT transform) |
| |
| template<typename InT , typename OutT , typename TransformT > |
| void | lbann::fft::r2c_to_full_2d (El::Matrix< InT, El::Device::CPU > const &r2c_input, El::Matrix< OutT, El::Device::CPU > &full_output, std::vector< int > const &full_dims, TransformT transform) |
| |
| template<typename InT , typename OutT > |
| void | lbann::fft::r2c_to_full (El::Matrix< InT, El::Device::CPU > const &r2c_input, El::Matrix< OutT, El::Device::CPU > &full_output, std::vector< int > const &full_dims) |
| |