Searched refs:ThenFft (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | fft_thunk.cc | 187 launch_ok = stream.ThenFft(fft_plan.get(), input_data, &output_data).ok(); in ExecuteOnStream()
195 launch_ok = stream.ThenFft(fft_plan.get(), input_data, &output_data).ok(); in ExecuteOnStream()
203 launch_ok = stream.ThenFft(fft_plan.get(), input_data, &output_data).ok(); in ExecuteOnStream()
217 launch_ok = stream.ThenFft(fft_plan.get(), input_data, &output_data).ok(); in ExecuteOnStream()
232 launch_ok = stream.ThenFft(fft_plan.get(), input_data, &output_data).ok(); in ExecuteOnStream()
240 launch_ok = stream.ThenFft(fft_plan.get(), input_data, &output_data).ok(); in ExecuteOnStream()
248 launch_ok = stream.ThenFft(fft_plan.get(), input_data, &output_data).ok(); in ExecuteOnStream()
262 launch_ok = stream.ThenFft(fft_plan.get(), input_data, &output_data).ok(); in ExecuteOnStream()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream.h | 1668 Stream &ThenFft(fft::Plan *plan,
1671 Stream &ThenFft(fft::Plan *plan,
1674 Stream &ThenFft(fft::Plan *plan, const DeviceMemory<float> &input,
1676 Stream &ThenFft(fft::Plan *plan, const DeviceMemory<double> &input,
1678 Stream &ThenFft(fft::Plan *plan,
1681 Stream &ThenFft(fft::Plan *plan,
|
| D | stream.cc | 4862 Stream &Stream::ThenFft(fft::Plan *plan, in ThenFft() function in stream_executor::Stream
4878 Stream &Stream::ThenFft(fft::Plan *plan, in ThenFft() function in stream_executor::Stream
4894 Stream &Stream::ThenFft(fft::Plan *plan, const DeviceMemory<float> &input, in ThenFft() function in stream_executor::Stream
4909 Stream &Stream::ThenFft(fft::Plan *plan, const DeviceMemory<double> &input, in ThenFft() function in stream_executor::Stream
4924 Stream &Stream::ThenFft(fft::Plan *plan, in ThenFft() function in stream_executor::Stream
4940 Stream &Stream::ThenFft(fft::Plan *plan, in ThenFft() function in stream_executor::Stream
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fft_ops.cc | 537 ctx, stream->ThenFft(plan, src, &dst).ok(), in DoFFTInternal()
|