Home
last modified time | relevance | path

Searched refs:FftFactory (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/
Dplugin_registry.h55 typedef fft::FftSupport* (*FftFactory)(internal::StreamExecutorInterface*); typedef
110 std::map<PluginId, FftFactory> fft;
Dplugin_registry.cc244 EMIT_PLUGIN_SPECIALIZATIONS(FftFactory, fft, "FFT");
/external/tensorflow/tensorflow/stream_executor/host/
Dhost_gpu_executor.cc227 ->GetFactory<PluginRegistry::FftFactory>(kHostPlatformId, in SupportsFft()
234 port::StatusOr<PluginRegistry::FftFactory> status = in CreateFft()
235 registry->GetFactory<PluginRegistry::FftFactory>(kHostPlatformId, in CreateFft()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_gpu_executor.cc685 port::StatusOr<PluginRegistry::FftFactory> status = in CreateFft()
686 registry->GetFactory<PluginRegistry::FftFactory>(rocm::kROCmPlatformId, in CreateFft()
Drocm_fft.cc591 PluginRegistry::Instance()->RegisterFactory<PluginRegistry::FftFactory>( in initialize_rocfft()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_gpu_executor.cc804 port::StatusOr<PluginRegistry::FftFactory> status = in CreateFft()
805 registry->GetFactory<PluginRegistry::FftFactory>(cuda::kCudaPlatformId, in CreateFft()
Dcuda_fft.cc538 PluginRegistry::Instance()->RegisterFactory<PluginRegistry::FftFactory>( in initialize_cufft()