Home
last modified time | relevance | path

Searched defs:gpu_version (Results 1 – 10 of 10) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/llvm_gpu_backend/
H A Dgpu_backend_lib.cc337 Status NVPTXTargetModuleLinker(llvm::Module* module, GpuVersion gpu_version, in NVPTXTargetModuleLinker()
371 Status LinkAndOptimizeModule(llvm::Module* module, GpuVersion gpu_version, in LinkAndOptimizeModule()
512 llvm::Module* module, GpuVersion gpu_version, in CompileToPtx()
788 Status AMDGPUTargetModuleLinker(llvm::Module* module, GpuVersion gpu_version, in AMDGPUTargetModuleLinker()
861 llvm::Triple target_triple, GpuVersion gpu_version, in AMDGPUGetTargetMachine()
895 llvm::Module* module, GpuVersion gpu_version, in CompileToHsaco()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dgpu_compiler.cc1130 GpuVersion gpu_version = GetGpuVersion(stream_exec); in CompileToTargetBinary() local
1402 GpuVersion gpu_version = GetGpuVersion(stream_exec); in RunBackend() local
1654 GpuVersion gpu_version = compiler->GetGpuVersion(stream_exec); in CompileLmhloToExecutable() local
H A Damdgpu_compiler.cc136 GpuVersion gpu_version, in CompileTargetBinary()
H A Dgpu_executable.cc1007 std::shared_ptr<HloModule> hlo_module, GpuVersion gpu_version, in GpuExecutable()
1029 GpuVersion gpu_version, se::StreamExecutor* executor) { in LoadFromObjFile()
H A Dgpu_executable.h108 GpuVersion gpu_version; member
H A Dnvptx_compiler.cc330 GpuVersion gpu_version, in CompileTargetBinary()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dllvm_compiler_test.cc64 GpuVersion gpu_version, se::StreamExecutor* stream_exec, bool relocatable, in CompileTargetBinary()
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/cl/
H A Dcl_device.cc160 bool IsGPUVersionInRange(int gpu_version, int min_version, int max_version) { in IsGPUVersionInRange()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/lmhlo_to_gpu/
H A Dkernel_ops_pattern.cc311 xla::gpu::GpuVersion gpu_version{rocm_compute_capability}; in Match() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/
H A Dgpu_info.h241 MaliGpu gpu_version; member