Home
last modified time | relevance | path

Searched defs:fft_rank (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/python/ops/signal/
Dfft_ops.py31 def _infer_fft_length_for_rfft(input_tensor, fft_rank): argument
44 def _infer_fft_length_for_irfft(input_tensor, fft_rank): argument
62 def _maybe_pad_for_rfft(input_tensor, fft_rank, fft_length, is_reverse=False): argument
110 def _rfft_wrapper(fft_fn, fft_rank, default_name): argument
129 def _irfft_wrapper(ifft_fn, fft_rank, default_name): argument
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Druntime_single_threaded_fft.cc27 int32 fft_rank, int64 input_batch, int64 fft_length0, int64 fft_length1, in __xla_cpu_runtime_EigenSingleThreadedFft()
Druntime_fft.cc30 int32 fft_rank, int64 input_batch, int64 fft_length0, int64 fft_length1, in __xla_cpu_runtime_EigenFft()
Druntime_fft_impl.h208 int32 fft_type, int32 fft_rank, int64 input_batch, in EigenFftImpl()
Dir_emitter.cc1296 const int fft_rank = fft_length.size(); in HandleFft() local
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dfft_ops.cc43 int fft_rank) in GenericFftOp()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dfft_thunk.cc122 const int64 fft_rank = fft_length_.size(); in ExecuteOnStream() local
/external/tensorflow/tensorflow/core/kernels/
Dfft_ops.cc44 const int fft_rank = Rank(); in Compute() local
384 const int fft_rank = Rank(); in DoFFT() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference.cc1821 const int64 fft_rank = fft_length.size(); in InferFftShape() local