fft.h File Reference

#include <variant>
#include "array.h"
#include "device.h"
#include "utils.h"

Go to the source code of this file.

Namespaces
namespace	mlx
namespace	mlx::core
namespace	mlx::core::fft

Functions
array	mlx::core::fft::fftn (const array &a, const std::vector< int > &n, const std::vector< int > &axes, StreamOrDevice s={})
	Compute the n-dimensional Fourier Transform.
array	mlx::core::fft::fftn (const array &a, const std::vector< int > &axes, StreamOrDevice s={})
array	mlx::core::fft::fftn (const array &a, StreamOrDevice s={})
array	mlx::core::fft::ifftn (const array &a, const std::vector< int > &n, const std::vector< int > &axes, StreamOrDevice s={})
	Compute the n-dimensional inverse Fourier Transform.
array	mlx::core::fft::ifftn (const array &a, const std::vector< int > &axes, StreamOrDevice s={})
array	mlx::core::fft::ifftn (const array &a, StreamOrDevice s={})
array	mlx::core::fft::fft (const array &a, int n, int axis, StreamOrDevice s={})
	Compute the one-dimensional Fourier Transform.
array	mlx::core::fft::fft (const array &a, int axis=-1, StreamOrDevice s={})
array	mlx::core::fft::ifft (const array &a, int n, int axis, StreamOrDevice s={})
	Compute the one-dimensional inverse Fourier Transform.
array	mlx::core::fft::ifft (const array &a, int axis=-1, StreamOrDevice s={})
array	mlx::core::fft::fft2 (const array &a, const std::vector< int > &n, const std::vector< int > &axes, StreamOrDevice s={})
	Compute the two-dimensional Fourier Transform.
array	mlx::core::fft::fft2 (const array &a, const std::vector< int > &axes={-2, -1}, StreamOrDevice s={})
array	mlx::core::fft::ifft2 (const array &a, const std::vector< int > &n, const std::vector< int > &axes, StreamOrDevice s={})
	Compute the two-dimensional inverse Fourier Transform.
array	mlx::core::fft::ifft2 (const array &a, const std::vector< int > &axes={-2, -1}, StreamOrDevice s={})
array	mlx::core::fft::rfftn (const array &a, const std::vector< int > &n, const std::vector< int > &axes, StreamOrDevice s={})
	Compute the n-dimensional Fourier Transform on a real input.
array	mlx::core::fft::rfftn (const array &a, const std::vector< int > &axes, StreamOrDevice s={})
array	mlx::core::fft::rfftn (const array &a, StreamOrDevice s={})
array	mlx::core::fft::irfftn (const array &a, const std::vector< int > &n, const std::vector< int > &axes, StreamOrDevice s={})
	Compute the n-dimensional inverse of rfftn.
array	mlx::core::fft::irfftn (const array &a, const std::vector< int > &axes, StreamOrDevice s={})
array	mlx::core::fft::irfftn (const array &a, StreamOrDevice s={})
array	mlx::core::fft::rfft (const array &a, int n, int axis, StreamOrDevice s={})
	Compute the one-dimensional Fourier Transform on a real input.
array	mlx::core::fft::rfft (const array &a, int axis=-1, StreamOrDevice s={})
array	mlx::core::fft::irfft (const array &a, int n, int axis, StreamOrDevice s={})
	Compute the one-dimensional inverse of rfft.
array	mlx::core::fft::irfft (const array &a, int axis=-1, StreamOrDevice s={})
array	mlx::core::fft::rfft2 (const array &a, const std::vector< int > &n, const std::vector< int > &axes, StreamOrDevice s={})
	Compute the two-dimensional Fourier Transform on a real input.
array	mlx::core::fft::rfft2 (const array &a, const std::vector< int > &axes={-2, -1}, StreamOrDevice s={})
array	mlx::core::fft::irfft2 (const array &a, const std::vector< int > &n, const std::vector< int > &axes, StreamOrDevice s={})
	Compute the two-dimensional inverse of rfft2.
array	mlx::core::fft::irfft2 (const array &a, const std::vector< int > &axes={-2, -1}, StreamOrDevice s={})