Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Damdgpu_compiler.cc97 if (!stream_exec->GetDeviceDescription().rocm_amdgpu_isa_version( in GetGpuVersion()
Dgpu_executable.cc102 main_stream->parent()->GetDeviceDescription().rocm_amdgpu_isa_version( in CheckCompatibilityWithServiceExecutableRunOptions()
/external/tensorflow/tensorflow/stream_executor/
Ddevice_description.cc120 bool DeviceDescription::rocm_amdgpu_isa_version(int *version) const { in rocm_amdgpu_isa_version() function in stream_executor::DeviceDescription
Ddevice_description.h139 bool rocm_amdgpu_isa_version(int *version) const;
/external/tensorflow/tensorflow/compiler/xla/service/
Dplatform_util.cc150 if (description.rocm_amdgpu_isa_version(&isa_version)) { in IsDeviceSupported()
/external/tensorflow/tensorflow/stream_executor/tpu/
Dc_api_decl.h151 int rocm_amdgpu_isa_version; member
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_device.cc1872 if (!desc->rocm_amdgpu_isa_version(&device_isa)) { in GetValidDeviceIds()