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.cc442 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.cc308 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.cc86 void KernelBase::SetPreferredCacheConfig(KernelCacheConfig config) { in SetPreferredCacheConfig()
90 KernelCacheConfig KernelBase::GetPreferredCacheConfig() const { in GetPreferredCacheConfig()
Dkernel_cache_config.h26 enum class KernelCacheConfig { enum
Dstream_executor_internal.h101 virtual void SetPreferredCacheConfig(KernelCacheConfig config) = 0;
104 virtual KernelCacheConfig GetPreferredCacheConfig() const = 0;
Dkernel.h175 void SetPreferredCacheConfig(KernelCacheConfig config);
178 KernelCacheConfig GetPreferredCacheConfig() const;