Searched refs:GpuVersion (Results 1 – 13 of 13) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/llvm_gpu_backend/ |
| D | gpu_backend_lib.h | 43 llvm::Module* module, GpuVersion gpu_version,
53 llvm::Module* module, GpuVersion gpu_version,
|
| D | gpu_backend_lib.cc | 338 Status NVPTXTargetModuleLinker(llvm::Module* module, GpuVersion gpu_version, in NVPTXTargetModuleLinker()
375 llvm::Module*, GpuVersion, const HloModuleConfig&, const string&)>;
377 Status LinkAndOptimizeModule(llvm::Module* module, GpuVersion gpu_version, in LinkAndOptimizeModule()
513 llvm::Module* module, GpuVersion gpu_version, in CompileToPtx()
786 Status AMDGPUTargetModuleLinker(llvm::Module* module, GpuVersion gpu_version, in AMDGPUTargetModuleLinker()
855 llvm::Triple target_triple, GpuVersion gpu_version, in AMDGPUGetTargetMachine()
887 llvm::Module* module, GpuVersion gpu_version, in CompileToHsaco()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | amdgpu_compiler.h | 40 GpuVersion GetGpuVersion(se::StreamExecutor* stream_exec) override;
44 GpuVersion gpu_version, se::StreamExecutor* stream_exec, bool relocatable,
|
| D | amdgpu_compiler.cc | 95 GpuVersion AMDGPUCompiler::GetGpuVersion(se::StreamExecutor* stream_exec) { in GetGpuVersion()
116 GpuVersion gpu_version, in CompileTargetBinary()
|
| D | nvptx_compiler.h | 51 GpuVersion GetGpuVersion(se::StreamExecutor* stream_exec) override;
55 GpuVersion gpu_version, se::StreamExecutor* stream_exec, bool relocatable,
|
| D | gpu_types.h | 38 using GpuVersion = variable
|
| D | gpu_executable.h | 73 GpuVersion gpu_version;
190 GpuVersion gpu_version_;
|
| D | gpu_compiler.h | 84 virtual GpuVersion GetGpuVersion(se::StreamExecutor* stream_exec) = 0;
90 llvm::Module* llvm_module, GpuVersion gpu_version,
|
| D | nvptx_compiler.cc | 287 GpuVersion NVPTXCompiler::GetGpuVersion(se::StreamExecutor* stream_exec) { in GetGpuVersion()
303 GpuVersion gpu_version, in CompileTargetBinary()
|
| D | gpu_compiler.cc | 708 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()
|
| D | gpu_executable.cc | 114 GpuVersion nvidia_compute_compatibility = stream_compute_compatibility; in CheckCompatibilityWithServiceExecutableRunOptions()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | llvm_compiler_test.cc | 56 GpuVersion GetGpuVersion(se::StreamExecutor*) override { in GetGpuVersion()
62 GpuVersion gpu_version, se::StreamExecutor* stream_exec, bool relocatable, in CompileTargetBinary()
|
| /external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/ |
| D | gpu_kernel_to_blob_pass.cc | 123 xla::gpu::GpuVersion gpu_version{std::make_pair(arch, arch_str)}; in GetGpuBinaryBlob()
|