Searched defs:compute_capability (Results 1 – 3 of 3) 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
114 static string GetSmName(std::pair<int, int> compute_capability) { in GetSmName() argument
301 std::pair<int, int> compute_capability, in LinkLibdeviceIfNecessary() argument
329 std::pair<int, int> compute_capability, in CompileModuleToPtx() argument
483 std::pair<int, int> compute_capability, in CompileToPtx() argument
|
| /external/tensorflow/tensorflow/core/protobuf/ |
| D | autotuning.proto | 72 ComputeCapability compute_capability = 4; field
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_executable.cc | 52 std::pair<int, int> compute_capability, in GpuExecutable() argument
|