MLX
 
Loading...
Searching...
No Matches
pocketfft Namespace Reference

Namespaces

namespace  detail
 

Typedefs

using shape_t
 
using stride_t
 

Functions

template<typename T>
void c2c (const shape_t &shape, const stride_t &stride_in, const stride_t &stride_out, const shape_t &axes, bool forward, const std::complex< T > *data_in, std::complex< T > *data_out, T fct, size_t nthreads=1)
 
template<typename T>
void c2r (const shape_t &shape_out, const stride_t &stride_in, const stride_t &stride_out, size_t axis, bool forward, const std::complex< T > *data_in, T *data_out, T fct, size_t nthreads=1)
 
template<typename T>
void r2c (const shape_t &shape_in, const stride_t &stride_in, const stride_t &stride_out, size_t axis, bool forward, const T *data_in, std::complex< T > *data_out, T fct, size_t nthreads=1)
 
template<typename T>
void r2r_fftpack (const shape_t &shape, const stride_t &stride_in, const stride_t &stride_out, const shape_t &axes, bool real2hermitian, bool forward, const T *data_in, T *data_out, T fct, size_t nthreads=1)
 
template<typename T>
void r2r_separable_hartley (const shape_t &shape, const stride_t &stride_in, const stride_t &stride_out, const shape_t &axes, const T *data_in, T *data_out, T fct, size_t nthreads=1)
 
template<typename T>
void r2r_genuine_hartley (const shape_t &shape, const stride_t &stride_in, const stride_t &stride_out, const shape_t &axes, const T *data_in, T *data_out, T fct, size_t nthreads=1)
 
template<typename T>
void dct (const shape_t &shape, const stride_t &stride_in, const stride_t &stride_out, const shape_t &axes, int type, const T *data_in, T *data_out, T fct, bool ortho, size_t nthreads=1)
 
template<typename T>
void dst (const shape_t &shape, const stride_t &stride_in, const stride_t &stride_out, const shape_t &axes, int type, const T *data_in, T *data_out, T fct, bool ortho, size_t nthreads=1)
 

Variables

constexpr bool FORWARD
 
constexpr bool BACKWARD
 

Typedef Documentation

◆ shape_t

◆ stride_t

Function Documentation

◆ c2c()

template<typename T>
void pocketfft::detail::c2c ( const shape_t & shape,
const stride_t & stride_in,
const stride_t & stride_out,
const shape_t & axes,
bool forward,
const std::complex< T > * data_in,
std::complex< T > * data_out,
T fct,
size_t nthreads = 1 )

◆ c2r()

template<typename T>
void pocketfft::detail::c2r ( const shape_t & shape_out,
const stride_t & stride_in,
const stride_t & stride_out,
size_t axis,
bool forward,
const std::complex< T > * data_in,
T * data_out,
T fct,
size_t nthreads = 1 )

◆ dct()

template<typename T>
void pocketfft::detail::dct ( const shape_t & shape,
const stride_t & stride_in,
const stride_t & stride_out,
const shape_t & axes,
int type,
const T * data_in,
T * data_out,
T fct,
bool ortho,
size_t nthreads = 1 )

◆ dst()

template<typename T>
void pocketfft::detail::dst ( const shape_t & shape,
const stride_t & stride_in,
const stride_t & stride_out,
const shape_t & axes,
int type,
const T * data_in,
T * data_out,
T fct,
bool ortho,
size_t nthreads = 1 )

◆ r2c()

template<typename T>
void pocketfft::detail::r2c ( const shape_t & shape_in,
const stride_t & stride_in,
const stride_t & stride_out,
size_t axis,
bool forward,
const T * data_in,
std::complex< T > * data_out,
T fct,
size_t nthreads = 1 )

◆ r2r_fftpack()

template<typename T>
void pocketfft::detail::r2r_fftpack ( const shape_t & shape,
const stride_t & stride_in,
const stride_t & stride_out,
const shape_t & axes,
bool real2hermitian,
bool forward,
const T * data_in,
T * data_out,
T fct,
size_t nthreads = 1 )

◆ r2r_genuine_hartley()

template<typename T>
void pocketfft::detail::r2r_genuine_hartley ( const shape_t & shape,
const stride_t & stride_in,
const stride_t & stride_out,
const shape_t & axes,
const T * data_in,
T * data_out,
T fct,
size_t nthreads = 1 )

◆ r2r_separable_hartley()

template<typename T>
void pocketfft::detail::r2r_separable_hartley ( const shape_t & shape,
const stride_t & stride_in,
const stride_t & stride_out,
const shape_t & axes,
const T * data_in,
T * data_out,
T fct,
size_t nthreads = 1 )

Variable Documentation

◆ BACKWARD

bool pocketfft::detail::BACKWARD

◆ FORWARD

bool pocketfft::detail::FORWARD
constexpr