Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/host/
Dhost_gpu_executor.cc44 : plugin_config_(plugin_config) {} in HostExecutor()
299 plugin_config_.blas()) in SupportsBlas()
307 plugin_config_.blas()); in CreateBlas()
320 plugin_config_.fft()) in SupportsFft()
328 plugin_config_.fft()); in CreateFft()
341 plugin_config_.rng()) in SupportsRng()
349 plugin_config_.rng()); in CreateRng()
Dhost_gpu_executor.h180 const PluginConfig plugin_config_;
/external/tensorflow/tensorflow/stream_executor/gpu/
Dgpu_executor.h55 plugin_config_(plugin_config) {} in GpuExecutor()
338 PluginConfig plugin_config_; variable
/external/tensorflow/tensorflow/compiler/xla/service/interpreter/
Dexecutor.cc32 : plugin_config_(plugin_config) {} in XlaInterpreterExecutor()
Dexecutor.h210 const PluginConfig plugin_config_; variable
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_gpu_executor.cc658 plugin_config_.blas()); in CreateBlas()
672 plugin_config_.dnn()); in CreateDnn()
686 plugin_config_.fft()); in CreateFft()
700 plugin_config_.rng()); in CreateRng()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_gpu_executor.cc750 plugin_config_.blas()); in CreateBlas()
764 plugin_config_.dnn()); in CreateDnn()
778 plugin_config_.fft()); in CreateFft()
792 plugin_config_.rng()); in CreateRng()