Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/
H A 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()
H A Drocm_gpu_executor.cc321 KernelCacheConfig::kNoPreference) { in Launch()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
H A 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()
H A Dcuda_gpu_executor.cc464 KernelCacheConfig::kNoPreference) { in Launch()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/gpu/
H A 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_;
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/
H A Dkernel.cc85 void KernelBase::SetPreferredCacheConfig(KernelCacheConfig config) { in SetPreferredCacheConfig()
89 KernelCacheConfig KernelBase::GetPreferredCacheConfig() const { in GetPreferredCacheConfig()
H A Dstream_executor_internal.h100 virtual void SetPreferredCacheConfig(KernelCacheConfig config) = 0;
103 virtual KernelCacheConfig GetPreferredCacheConfig() const = 0;
H A Dkernel_cache_config.h26 enum class KernelCacheConfig { enum
H A Dkernel.h176 void SetPreferredCacheConfig(KernelCacheConfig config);
179 KernelCacheConfig GetPreferredCacheConfig() const;