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 | 315 Status NVPTXTargetModuleLinker(llvm::Module* module, GpuVersion gpu_version, in NVPTXTargetModuleLinker()
354 Status LinkAndOptimizeModule(llvm::Module* module, GpuVersion gpu_version, in LinkAndOptimizeModule()
489 StatusOr<string> CompileToPtx(llvm::Module* module, GpuVersion gpu_version, in CompileToPtx()
667 Status AMDGPUTargetModuleLinker(llvm::Module* module, GpuVersion gpu_version, in AMDGPUTargetModuleLinker()
710 llvm::Module* module, GpuVersion gpu_version, in CompileToHsaco()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | amdgpu_compiler.cc | 134 GpuVersion gpu_version, in CompileTargetBinary()
|
| D | gpu_compiler.cc | 420 GpuVersion gpu_version = GetGpuVersion(stream_exec); in RunBackend() local
|
| D | nvptx_compiler.cc | 342 GpuVersion gpu_version, in CompileTargetBinary()
|
| D | gpu_executable.cc | 55 GpuVersion gpu_version, std::unique_ptr<const ThunkSchedule> thunk_schedule, in GpuExecutable()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/ |
| D | cl_device.cc | 141 bool IsGPUVersionInRange(int gpu_version, int min_version, int max_version) { in IsGPUVersionInRange()
159 int GetAdrenoGPUVersion(const std::string& gpu_version) { in GetAdrenoGPUVersion()
|
| D | cl_device.h | 43 int gpu_version = -1; // can be, for example, 405/430/540/530/630 etc. member
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/ |
| D | conv_constants.cc | 173 int GetAdrenoOptimalMaxConstantSize(int gpu_version) { in GetAdrenoOptimalMaxConstantSize()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | llvm_compiler_test.cc | 60 GpuVersion gpu_version, se::StreamExecutor* stream_exec) { in CompileTargetBinary()
|