Searched refs:compute_capability (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/llvm_gpu_backend/ |
| D | nvptx_backend_lib.cc | 75 std::pair<int, int> compute_capability) { in GetLibdeviceFilename() argument
101 auto it = m->find(compute_capability); in GetLibdeviceFilename()
105 LOG(WARNING) << "Unknown compute capability (" << compute_capability.first in GetLibdeviceFilename()
106 << ", " << compute_capability.second << ") ." in GetLibdeviceFilename()
114 static string GetSmName(std::pair<int, int> compute_capability) { in GetSmName() argument
129 auto it = m->find(compute_capability); in GetSmName()
133 LOG(WARNING) << "Unknown compute capability (" << compute_capability.first in GetSmName()
134 << ", " << compute_capability.second << ") ." in GetSmName()
301 std::pair<int, int> compute_capability, in LinkLibdeviceIfNecessary() argument
310 GetLibdeviceFilename(libdevice_dir_path, compute_capability)); in LinkLibdeviceIfNecessary()
[all …]
|
| D | nvptx_backend_lib.h | 40 std::pair<int, int> compute_capability,
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_executable.h | 55 std::pair<int, int> compute_capability,
|
| D | gpu_executable.cc | 52 std::pair<int, int> compute_capability, in GpuExecutable() argument
62 compute_capability_(compute_capability), in GpuExecutable()
|
| /external/tensorflow/tensorflow/core/protobuf/ |
| D | autotuning.proto | 72 ComputeCapability compute_capability = 4; field
|
| /external/tensorflow/ |
| D | configure.py | 1303 for compute_capability in tf_cuda_compute_capabilities.split(','):
1304 m = re.match('[0-9]+.[0-9]+', compute_capability)
1306 print('Invalid compute capability: %s' % compute_capability)
|