Searched refs:ThenFft (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | fft_thunk.cc | 165 stream->ThenFft(fft_plan_.get(), input_data, &output_data).ok(); in ExecuteOnStream()
174 stream->ThenFft(fft_plan_.get(), input_data, &output_data).ok(); in ExecuteOnStream()
190 stream->ThenFft(fft_plan_.get(), input_data, &output_data).ok(); in ExecuteOnStream()
199 stream->ThenFft(fft_plan_.get(), input_data, &output_data).ok(); in ExecuteOnStream()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fft_ops.cc | 428 ctx, stream->ThenFft(plan.get(), src, &dst).ok(), in DoFFT()
435 ctx, stream->ThenFft(plan.get(), src, &dst).ok(), in DoFFT()
453 ctx, stream->ThenFft(plan.get(), src, &dst).ok(), in DoFFT()
471 ctx, stream->ThenFft(plan.get(), src, &dst).ok(), in DoFFT()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream.h | 1626 Stream &ThenFft(fft::Plan *plan,
1629 Stream &ThenFft(fft::Plan *plan,
1632 Stream &ThenFft(fft::Plan *plan, const DeviceMemory<float> &input,
1634 Stream &ThenFft(fft::Plan *plan, const DeviceMemory<double> &input,
1636 Stream &ThenFft(fft::Plan *plan,
1639 Stream &ThenFft(fft::Plan *plan,
|
| D | stream.cc | 5139 Stream &Stream::ThenFft(fft::Plan *plan, in ThenFft() function in stream_executor::Stream
5157 Stream &Stream::ThenFft(fft::Plan *plan, in ThenFft() function in stream_executor::Stream
5175 Stream &Stream::ThenFft(fft::Plan *plan, const DeviceMemory<float> &input, in ThenFft() function in stream_executor::Stream
5192 Stream &Stream::ThenFft(fft::Plan *plan, const DeviceMemory<double> &input, in ThenFft() function in stream_executor::Stream
5209 Stream &Stream::ThenFft(fft::Plan *plan, in ThenFft() function in stream_executor::Stream
5227 Stream &Stream::ThenFft(fft::Plan *plan, in ThenFft() function in stream_executor::Stream
|