Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/
Dplugin_registry.h54 typedef fft::FftSupport* (*FftFactory)(internal::StreamExecutorInterface*); typedef
109 std::map<PluginId, FftFactory> fft;
178 DECLARE_PLUGIN_SPECIALIZATIONS(FftFactory);
Dplugin_registry.cc248 EMIT_PLUGIN_SPECIALIZATIONS(FftFactory, fft, "FFT");
/external/tensorflow/tensorflow/stream_executor/host/
Dhost_gpu_executor.cc319 ->GetFactory<PluginRegistry::FftFactory>(kHostPlatformId, in SupportsFft()
326 port::StatusOr<PluginRegistry::FftFactory> status = in CreateFft()
327 registry->GetFactory<PluginRegistry::FftFactory>(kHostPlatformId, in CreateFft()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_gpu_executor.cc684 port::StatusOr<PluginRegistry::FftFactory> status = in CreateFft()
685 registry->GetFactory<PluginRegistry::FftFactory>(rocm::kROCmPlatformId, in CreateFft()
Drocm_fft.cc596 PluginRegistry::Instance()->RegisterFactory<PluginRegistry::FftFactory>( in initialize_rocfft()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_gpu_executor.cc776 port::StatusOr<PluginRegistry::FftFactory> status = in CreateFft()
777 registry->GetFactory<PluginRegistry::FftFactory>(cuda::kCudaPlatformId, in CreateFft()
Dcuda_fft.cc567 PluginRegistry::Instance()->RegisterFactory<PluginRegistry::FftFactory>( in initialize_cufft()