Home
last modified time | relevance | path

Searched refs:AsFft (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dfft_thunk.cc170 fft_plan = stream.parent()->AsFft()->CreateBatchedPlanWithScratchAllocator( in ExecuteOnStream()
176 stream.parent()->AsFft()->UpdatePlanWithScratchAllocator( in ExecuteOnStream()
/external/tensorflow/tensorflow/stream_executor/
Dstream_executor_pimpl.h485 fft::FftSupport *AsFft();
Dstream_executor_pimpl.cc444 fft::FftSupport *StreamExecutor::AsFft() { in AsFft() function in stream_executor::StreamExecutor
Dstream.cc4867 if (fft::FftSupport *fft = parent_->AsFft()) { in ThenFft()
4883 if (fft::FftSupport *fft = parent_->AsFft()) { in ThenFft()
4898 if (fft::FftSupport *fft = parent_->AsFft()) { in ThenFft()
4913 if (fft::FftSupport *fft = parent_->AsFft()) { in ThenFft()
4929 if (fft::FftSupport *fft = parent_->AsFft()) { in ThenFft()
4945 if (fft::FftSupport *fft = parent_->AsFft()) { in ThenFft()
/external/tensorflow/tensorflow/core/kernels/
Dfft_ops.cc470 stream->parent()->AsFft()->CreateBatchedPlanWithScratchAllocator( in DoFFT()