Home
last modified time | relevance | path

Searched refs:FftSupport (Results 1 – 15 of 15) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/
Dfft.h89 class FftSupport {
91 virtual ~FftSupport() {} in ~FftSupport()
204 FftSupport() {} in FftSupport() function
207 SE_DISALLOW_COPY_AND_ASSIGN(FftSupport);
Dplugin_registry.h55 typedef fft::FftSupport* (*FftFactory)(internal::StreamExecutorInterface*);
Dstream_executor_pimpl.h463 fft::FftSupport *AsFft();
666 std::unique_ptr<fft::FftSupport> fft_;
Dstream_executor_internal.h330 virtual fft::FftSupport *CreateFft() { return nullptr; } in CreateFft()
Dstream_executor_pimpl.cc459 fft::FftSupport *StreamExecutor::AsFft() { in AsFft()
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/stream_executor/rocm/
Drocm_fft.h100 class ROCMFft : public fft::FftSupport {
Drocm_fft.cc593 [](internal::StreamExecutorInterface *parent) -> fft::FftSupport * { in initialize_rocfft()
Drocm_gpu_executor.cc683 fft::FftSupport* GpuExecutor::CreateFft() { in CreateFft()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_fft.h105 class CUDAFft : public fft::FftSupport {
Dcuda_fft.cc540 [](internal::StreamExecutorInterface *parent) -> fft::FftSupport * { in initialize_cufft()
Dcuda_gpu_executor.cc802 fft::FftSupport* GpuExecutor::CreateFft() { in CreateFft()
/external/tensorflow/tensorflow/stream_executor/host/
Dhost_gpu_executor.h181 fft::FftSupport *CreateFft() override;
Dhost_gpu_executor.cc232 fft::FftSupport *HostExecutor::CreateFft() { in CreateFft()
/external/tensorflow/tensorflow/stream_executor/gpu/
Dgpu_executor.h214 fft::FftSupport* CreateFft() override;