Searched defs:fft_rank (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/signal/ |
| 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
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | runtime_single_threaded_fft.cc | 27 int32 double_precision, int32 fft_rank, int64 input_batch, in __xla_cpu_runtime_EigenSingleThreadedFft()
|
| D | runtime_fft.cc | 31 int32 double_precision, int32 fft_rank, int64 input_batch, in __xla_cpu_runtime_EigenFft()
|
| D | runtime_fft_impl.h | 245 FftType fft_type, bool double_precision, int32 fft_rank, in EigenFftImpl()
|
| D | ir_emitter.cc | 1047 const int fft_rank = fft_length.size(); in HandleFft() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | fft_ops.cc | 43 int fft_rank) in GenericFftOp()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fft_ops.cc | 46 const int fft_rank = Rank(); in Compute() local
433 const int fft_rank = Rank(); in DoFFT() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | fft_thunk.cc | 146 const int64 fft_rank = fft_length_.size(); in ExecuteOnStream() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_evaluator.cc | 999 void Sweep(int64 fft_rank, FftType fft_type, in Sweep()
1089 const absl::Span<const int64> src_strides, int64 fft_rank, in GenerateIndices()
1131 int64 fft_rank, FftType fft_type, int64 fft_size, in CopyDataFromInput()
1184 int64 fft_rank, FftType fft_type, in CopyDataToOutput()
1221 int64 fft_rank, FftType fft_type, int64 fft_size, in CopyDataFromInput()
1241 int64 fft_rank, FftType fft_type, in CopyDataToOutput()
1260 int64 fft_rank, FftType fft_type, in CheckParameters()
1377 const int64 fft_rank = fft_lengths.size(); in HandleFft() local
|
| D | shape_inference.cc | 1876 const int64 fft_rank = fft_length.size(); in InferFftShape() local
|