Searched defs:compute_capability (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/llvm_gpu_backend/ |
| D | gpu_backend_lib.cc | 76 std::pair<int, int> compute_capability) { in GetLibdeviceFilename() argument
117 static string GetSmName(std::pair<int, int> compute_capability) { in GetSmName() argument
315 llvm::Module* module, std::pair<int, int> compute_capability, in LinkLibdeviceIfNecessary() argument
343 std::pair<int, int> compute_capability, in CompileModuleToPtx() argument
486 std::pair<int, int> compute_capability, in CompileToPtx() argument
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_executable.cc | 115 std::pair<int, int> compute_capability, in GpuExecutable() argument
|