LBANN  0.103.0
LivermoreBigArtificialNeuralNetworkToolkit
fft_common.hpp File Reference
Include dependency graph for fft_common.hpp:
This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes

struct  lbann::ToRealT< T >
 
struct  lbann::ToRealT< El::Complex< T > >
 
struct  lbann::ToComplexT< T >
 
struct  lbann::ToComplexT< El::Complex< T > >
 
struct  lbann::fft::DimsHelper< InT, OutT >
 
struct  lbann::fft::DimsHelper< InOutT, InOutT >
 
struct  lbann::fft::DimsHelper< RealT, El::Complex< RealT > >
 
struct  lbann::fft::DimsHelper< El::Complex< RealT >, RealT >
 

Namespaces

 lbann
 
 lbann::fft
 

Typedefs

template<typename T >
using lbann::ToReal = typename ToRealT< T >::type
 
template<typename T >
using lbann::ToComplex = typename ToComplexT< T >::type
 

Functions

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)