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.h54 typedef fft::FftSupport* (*FftFactory)(internal::StreamExecutorInterface*);
Dstream_executor_pimpl.h485 fft::FftSupport *AsFft();
692 std::unique_ptr<fft::FftSupport> fft_;
Dstream_executor_internal.h333 virtual fft::FftSupport *CreateFft() { return nullptr; } in CreateFft()
Dstream_executor_pimpl.cc444 fft::FftSupport *StreamExecutor::AsFft() { in AsFft()
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/stream_executor/rocm/
Drocm_fft.h105 class ROCMFft : public fft::FftSupport {
Drocm_fft.cc598 [](internal::StreamExecutorInterface *parent) -> fft::FftSupport * { in initialize_rocfft()
Drocm_gpu_executor.cc682 fft::FftSupport* GpuExecutor::CreateFft() { in CreateFft()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_fft.h109 class CUDAFft : public fft::FftSupport {
Dcuda_fft.cc569 [](internal::StreamExecutorInterface *parent) -> fft::FftSupport * { in initialize_cufft()
Dcuda_gpu_executor.cc774 fft::FftSupport* GpuExecutor::CreateFft() { in CreateFft()
/external/tensorflow/tensorflow/stream_executor/host/
Dhost_gpu_executor.h158 fft::FftSupport *CreateFft() override;
Dhost_gpu_executor.cc324 fft::FftSupport *HostExecutor::CreateFft() { in CreateFft()
/external/tensorflow/tensorflow/stream_executor/gpu/
Dgpu_executor.h212 fft::FftSupport* CreateFft() override;