HomeSort by relevance Sort by last modified time
    Searched refs:fft_type_ (Results 1 - 7 of 7) sorted by null

  /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
cuda_fft.cc 131 fft_type_ = type;
309 switch (fft_type_) {
  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
fft_thunk.cc 118 fft_type_(FftTypeToSeType(fft_type)),
128 VLOG(3) << "FFT type: " << FftTypeToString(fft_type_);
165 fft_type_, kInPlaceFft, batch_size, &scratch_allocator);
173 switch (fft_type_) {
230 FftTypeToString(fft_type_).c_str());
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...]
hlo_instruction.cc 148 instruction->fft_type_ = proto.fft_type();
346 instruction->fft_type_ = fft_type;
    [all...]

Completed in 167 milliseconds