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.cc147 stream->parent()->AsFft()->CreateBatchedPlanWithScratchAllocator( in ExecuteOnStream()
153 stream->parent()->AsFft()->UpdatePlanWithScratchAllocator( in ExecuteOnStream()
/external/tensorflow/tensorflow/stream_executor/
Dstream_executor_pimpl.h463 fft::FftSupport *AsFft();
Dstream_executor_pimpl.cc459 fft::FftSupport *StreamExecutor::AsFft() { in AsFft() function in stream_executor::StreamExecutor
Dstream.cc5145 if (fft::FftSupport *fft = parent_->AsFft()) { in ThenFft()
5163 if (fft::FftSupport *fft = parent_->AsFft()) { in ThenFft()
5180 if (fft::FftSupport *fft = parent_->AsFft()) { in ThenFft()
5197 if (fft::FftSupport *fft = parent_->AsFft()) { in ThenFft()
5215 if (fft::FftSupport *fft = parent_->AsFft()) { in ThenFft()
5233 if (fft::FftSupport *fft = parent_->AsFft()) { in ThenFft()
/external/tensorflow/tensorflow/core/kernels/
Dfft_ops.cc418 stream->parent()->AsFft()->CreateBatchedPlanWithScratchAllocator( in DoFFT()