Searched defs:compute_capability (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/profiler/protobuf/ |
| D | hardware_types.proto | 29 CudaComputeCapability compute_capability = 5; field
|
| /external/tensorflow/tensorflow/python/keras/mixed_precision/ |
| D | device_compatibility_check_test.py | 29 def device_details(device_name, compute_capability=None): argument
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/llvm_gpu_backend/ |
| D | gpu_backend_lib.cc | 89 static string GetSmName(std::pair<int, int> compute_capability) { in GetSmName() argument
317 std::pair<int, int> compute_capability, in LinkLibdeviceIfNecessary() argument
343 auto compute_capability = absl::get_if<std::pair<int, int>>(&gpu_version); in NVPTXTargetModuleLinker() local
366 llvm::Triple target_triple, std::pair<int, int> compute_capability, in NVPTXGetTargetMachine() argument
535 auto compute_capability = absl::get_if<std::pair<int, int>>(&gpu_version); in CompileToPtx() local
|
| /external/tensorflow/tensorflow/core/protobuf/ |
| D | autotuning.proto | 80 ComputeCapability compute_capability = 4; field
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | nvptx_compiler.cc | 307 std::pair<int, int> compute_capability = in CompileTargetBinary() local
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | generic_layout_optimizer.cc | 63 double compute_capability = 0.0; in GetNumGPUs() local
|