Home
last modified time | relevance | path

Searched defs:rocm_compute_capability (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dir_emitter_context.h43 se::RocmComputeCapability rocm_compute_capability, in IrEmitterContext()
67 se::RocmComputeCapability rocm_compute_capability() const { in rocm_compute_capability() function
H A Dgpu_compiler.cc971 se::RocmComputeCapability rocm_compute_capability, in CompileModuleToLlvmIrImpl()
1525 se::RocmComputeCapability rocm_compute_capability, int pointer_size) { in CompileModuleToLlvmIr()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/
H A Ddevice_description.cc113 RocmComputeCapability DeviceDescription::rocm_compute_capability() const { in rocm_compute_capability() function in stream_executor::DeviceDescription
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
H A Dhlo_to_llvm_ir.cc70 tensorflow::se::RocmComputeCapability rocm_compute_capability("gfx908"); in CompileAndPrintLlvmIr() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dplatform_util.cc143 auto rocm_compute_capability = description.rocm_compute_capability(); in IsDeviceSupported() local
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/gpu/
H A Dgpu_device.cc1988 auto rocm_compute_capability = desc->rocm_compute_capability(); in GetValidDeviceIds() local