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)
|