Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Damdgpu_compiler.h40 GpuVersion GetGpuVersion(se::StreamExecutor* stream_exec) override;
Damdgpu_compiler.cc95 GpuVersion AMDGPUCompiler::GetGpuVersion(se::StreamExecutor* stream_exec) { in GetGpuVersion() function in xla::gpu::AMDGPUCompiler
Dnvptx_compiler.h51 GpuVersion GetGpuVersion(se::StreamExecutor* stream_exec) override;
Dgpu_compiler.h84 virtual GpuVersion GetGpuVersion(se::StreamExecutor* stream_exec) = 0;
Dgpu_compiler.cc708 GpuVersion gpu_version = GetGpuVersion(stream_exec); in CompileToTargetBinary()
948 GpuVersion gpu_version = GetGpuVersion(stream_exec); in RunBackend()
1143 GpuVersion gpu_version = compiler->GetGpuVersion(stream_exec); in CompileLmhloToExecutable()
Dnvptx_compiler.cc287 GpuVersion NVPTXCompiler::GetGpuVersion(se::StreamExecutor* stream_exec) { in GetGpuVersion() function in xla::gpu::NVPTXCompiler
/external/tensorflow/tensorflow/compiler/xla/tests/
Dllvm_compiler_test.cc56 GpuVersion GetGpuVersion(se::StreamExecutor*) override { in GetGpuVersion() function in xla::gpu::GpuDummyCompiler