Home
last modified time | relevance | path

Searched refs:KernelCacheConfig (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_kernel.cc23 case KernelCacheConfig::kNoPreference: in GetGpuCacheConfig()
25 case KernelCacheConfig::kPreferShared: in GetGpuCacheConfig()
27 case KernelCacheConfig::kPreferL1: in GetGpuCacheConfig()
29 case KernelCacheConfig::kPreferEqual: in GetGpuCacheConfig()
Dcuda_gpu_executor.cc423 KernelCacheConfig::kNoPreference) { in Launch()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_kernel.cc23 case KernelCacheConfig::kNoPreference: in GetGpuCacheConfig()
25 case KernelCacheConfig::kPreferShared: in GetGpuCacheConfig()
27 case KernelCacheConfig::kPreferL1: in GetGpuCacheConfig()
29 case KernelCacheConfig::kPreferEqual: in GetGpuCacheConfig()
Drocm_gpu_executor.cc319 KernelCacheConfig::kNoPreference) { in Launch()
/external/tensorflow/tensorflow/stream_executor/gpu/
Dgpu_kernel.h41 preferred_cache_config_(KernelCacheConfig::kNoPreference) {} in GpuKernel()
69 void SetPreferredCacheConfig(KernelCacheConfig config) override { in SetPreferredCacheConfig()
74 KernelCacheConfig GetPreferredCacheConfig() const override { in GetPreferredCacheConfig()
87 KernelCacheConfig preferred_cache_config_;
/external/tensorflow/tensorflow/stream_executor/
Dkernel.cc85 void KernelBase::SetPreferredCacheConfig(KernelCacheConfig config) { in SetPreferredCacheConfig()
89 KernelCacheConfig KernelBase::GetPreferredCacheConfig() const { in GetPreferredCacheConfig()
Dkernel_cache_config.h26 enum class KernelCacheConfig { enum
Dstream_executor_internal.h100 virtual void SetPreferredCacheConfig(KernelCacheConfig config) = 0;
103 virtual KernelCacheConfig GetPreferredCacheConfig() const = 0;
Dkernel.h176 void SetPreferredCacheConfig(KernelCacheConfig config);
179 KernelCacheConfig GetPreferredCacheConfig() const;