lbann::fft Namespace Reference

Classes
struct	DimsHelper
struct	DimsHelper< El::Complex< RealT >, RealT >
struct	DimsHelper< InOutT, InOutT >
struct	DimsHelper< RealT, El::Complex< RealT > >

Functions
template<typename T >
auto	get_r2c_output_dims (std::vector< T > const &dims)
template<typename InT , typename OutT , typename TransformT >
void	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	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	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)

Function Documentation

get_r2c_output_dims()

template<typename T >

auto lbann::fft::get_r2c_output_dims

(

std::vector< T > const &

dims

)

Definition at line 69 of file fft_common.hpp.

Here is the caller graph for this function:

r2c_to_full()

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
	)

Definition at line 262 of file fft_common.hpp.

Here is the call graph for this function:

r2c_to_full_1d()

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
	)

Definition at line 130 of file fft_common.hpp.

Here is the call graph for this function:

Here is the caller graph for this function:

r2c_to_full_2d()

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
	)

Definition at line 193 of file fft_common.hpp.

Here is the call graph for this function:

Here is the caller graph for this function: