Searched defs:fft_rank (Results 1 – 9 of 9) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/signal/ |
| H A D | fft_ops.py | 33 def _infer_fft_length_for_rfft(input_tensor, fft_rank): argument
46 def _infer_fft_length_for_irfft(input_tensor, fft_rank): argument
64 def _maybe_pad_for_rfft(input_tensor, fft_rank, fft_length, is_reverse=False): argument
112 def _rfft_wrapper(fft_fn, fft_rank, default_name): argument
146 def _irfft_wrapper(ifft_fn, fft_rank, default_name): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | fft_ops.cc | 47 const int fft_rank = Rank(); in Compute() local
444 const int fft_rank = Rank(); in DoFFT() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| H A D | runtime_single_threaded_fft.cc | 23 int32_t double_precision, int32_t fft_rank, int64_t input_batch, in __xla_cpu_runtime_EigenSingleThreadedFft()
|
| H A D | runtime_fft.cc | 27 int32_t double_precision, int32_t fft_rank, int64_t input_batch, in __xla_cpu_runtime_EigenFft()
|
| H A D | runtime_fft_impl.h | 242 int32_t fft_rank, int64_t input_batch, int64_t fft_length0, in EigenFftImpl()
|
| H A D | ir_emitter.cc | 1119 const int fft_rank = fft_length.size(); in HandleFft() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| H A D | fft_ops.cc | 44 int fft_rank) in GenericFftOp()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| H A D | fft_thunk.cc | 115 const int64_t fft_rank = fft_len.size(); in RunFft() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | shape_inference.cc | 1877 const int64_t fft_rank = fft_length.size(); in InferFftShape() local
|