org.nd4j.linalg.jcublas.fft

Class JcudaFft

java.lang.Object

org.nd4j.linalg.fft.BaseFFTInstance

org.nd4j.linalg.fft.DefaultFFTInstance

org.nd4j.linalg.jcublas.fft.JcudaFft

All Implemented Interfaces:

FFTInstance

public class JcudaFft
extends DefaultFFTInstance

Uses JCufft rather than the built in FFTs

Author:

Adam Gibson

Constructor Summary

Constructors

Constructor and Description
JcudaFft()

Method Summary

Modifier and Type	Method and Description
protected Op	getFftOp(INDArray arr, int n)
cufftHandle	getHandle() Get the handle for the current thread

Methods inherited from class org.nd4j.linalg.fft.DefaultFFTInstance

fft, fft, getIfftOp, ifft, ifft, ifft, ifft, postProcess, preProcess, rawfft, rawifft

Methods inherited from class org.nd4j.linalg.fft.BaseFFTInstance

doFFt, doInnerFft, fft, fft, fft, fft, fftn, fftn, fftn, fftn, ifftn, ifftn, ifftn, ifftn, ifftn, irfft, irfft, irfftn, rawfftn, rawifft, rawifftn

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

JcudaFft

public JcudaFft()

Method Detail

getHandle

public cufftHandle getHandle()

Get the handle for the current thread

Returns:

the handle for the current thread

getFftOp

protected Op getFftOp(INDArray arr,
                      int n)

Overrides:

getFftOp in class DefaultFFTInstance