HomeSort by relevance Sort by last modified time
    Searched full:fft_plan_ (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
fft_thunk.cc 136 if (fft_plan_ == nullptr) {
161 fft_plan_ =
169 stream, fft_plan_.get(), &scratch_allocator);
180 stream->ThenFft(fft_plan_.get(), input_data, &output_data).ok();
189 stream->ThenFft(fft_plan_.get(), input_data, &output_data).ok();
205 stream->ThenFft(fft_plan_.get(), input_data, &output_data).ok();
214 stream->ThenFft(fft_plan_.get(), input_data, &output_data).ok();
fft_thunk.h 72 FftThunk(const FftThunk&) = delete; // Cannot share fft_plan_
73 FftThunk& operator=(const FftThunk&) = delete; // Cannot share fft_plan_
86 std::unique_ptr<perftools::gputools::fft::Plan> fft_plan_; member in class:xla::gpu::FftThunk

Completed in 102 milliseconds