Home
last modified time | relevance | path

Searched refs:gcn_arch_name (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Damdgpu_compiler.cc103 std::string gcn_arch_name = in GetGpuVersion() local
105 if (gcn_arch_name == stream_exec->GetDeviceDescription().kUndefinedString) { in GetGpuVersion()
107 gcn_arch_name = "gfx803"; in GetGpuVersion()
110 return std::make_pair(isa_version, gcn_arch_name); in GetGpuVersion()
/external/tensorflow/tensorflow/stream_executor/
Ddevice_description.h307 void set_rocm_amdgpu_gcn_arch_name(const std::string &gcn_arch_name) { in set_rocm_amdgpu_gcn_arch_name() argument
308 device_description_->rocm_amdgpu_gcn_arch_name_ = gcn_arch_name; in set_rocm_amdgpu_gcn_arch_name()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/llvm_gpu_backend/
Dgpu_backend_lib.cc824 std::string GetFeatureStrFromGCNArchName(const std::string& gcn_arch_name) { in GetFeatureStrFromGCNArchName() argument
837 std::vector<std::string> tokens = absl::StrSplit(gcn_arch_name, ':'); in GetFeatureStrFromGCNArchName()
859 std::string gcn_arch_name = amdgpu_version->second; in AMDGPUGetTargetMachine() local
860 std::string feature_str = GetFeatureStrFromGCNArchName(gcn_arch_name); in AMDGPUGetTargetMachine()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_gpu_executor.cc823 std::string gcn_arch_name; in CreateDeviceDescription() local
824 status = GpuDriver::GetGpuGCNArchName(device, &gcn_arch_name); in CreateDeviceDescription()
897 absl::StrCat("AMDGPU ISA version: ", gcn_arch_name)); in CreateDeviceDescription()
905 builder.set_rocm_amdgpu_gcn_arch_name(gcn_arch_name); in CreateDeviceDescription()
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_device.cc1783 std::string gcn_arch_name = description->rocm_amdgpu_gcn_arch_name(); in GetValidDeviceIds() local
1787 << " ROCm AMDGPU Arch: " << gcn_arch_name in GetValidDeviceIds()