Searched defs:gpu_version (Results 1 – 9 of 9) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/service/gpu/llvm_gpu_backend/ |
D | gpu_backend_lib.cc | 338 Status NVPTXTargetModuleLinker(llvm::Module* module, GpuVersion gpu_version, in NVPTXTargetModuleLinker() 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.cc | 116 GpuVersion gpu_version, in CompileTargetBinary()
|
D | gpu_compiler.cc | 708 GpuVersion gpu_version = GetGpuVersion(stream_exec); in CompileToTargetBinary() local 948 GpuVersion gpu_version = GetGpuVersion(stream_exec); in RunBackend() local 1143 GpuVersion gpu_version = compiler->GetGpuVersion(stream_exec); in CompileLmhloToExecutable() local
|
D | gpu_executable.h | 73 GpuVersion gpu_version; member
|
D | nvptx_compiler.cc | 303 GpuVersion gpu_version, 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() local
|
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/ |
D | cl_device.cc | 127 bool IsGPUVersionInRange(int gpu_version, int min_version, int max_version) { in IsGPUVersionInRange()
|
/external/tensorflow/tensorflow/compiler/xla/tests/ |
D | llvm_compiler_test.cc | 62 GpuVersion gpu_version, se::StreamExecutor* stream_exec, bool relocatable, in CompileTargetBinary()
|
/external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
D | gpu_info.h | 193 MaliGpu gpu_version; member
|