Home | Sort by relevance Sort by last modified time |
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ | |
fft_ops.cc | 43 : XlaOpKernel(ctx), fft_type_(fft_type), fft_rank_(fft_rank) {} 52 if (fft_type_ == FftType::RFFT || fft_type_ == FftType::IRFFT) { 67 b->Fft(ctx->Input(0), fft_type_, fft_length); 72 const FftType fft_type_; member in class:tensorflow::__anon38744::GenericFftOp |
/external/tensorflow/tensorflow/stream_executor/cuda/ | |
cuda_fft.h | 51 fft_type_(fft::Type::kInvalid), 89 fft::Type fft_type_; member in class:perftools::gputools::cuda::CUDAFftPlan |
/external/tensorflow/tensorflow/compiler/xla/service/gpu/ | |
fft_thunk.h | 81 const perftools::gputools::fft::Type fft_type_; member in class:xla::gpu::FftThunk |
/external/tensorflow/tensorflow/compiler/xla/service/ | |
hlo_instruction.h | 1395 FftType fft_type_ = FftType::FFT; member in class:xla::HloInstruction [all...] |