Home
last modified time | relevance | path

Searched defs:cuda_compute_capability (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_context.h43 absl::optional<CudaComputeCapability> cuda_compute_capability, in IrEmitterContext()
65 absl::optional<CudaComputeCapability> cuda_compute_capability() const { in cuda_compute_capability() function
Dgpu_compiler.cc552 absl::optional<CudaComputeCapability> cuda_compute_capability, in CompileModuleToLlvmIrImpl()
869 absl::optional<CudaComputeCapability> cuda_compute_capability = in RunBackend() local
871 CudaComputeCapability cuda_compute_capability; in RunBackend() local
987 absl::optional<CudaComputeCapability> cuda_compute_capability, in CompileModuleToLlvmIr()
Dir_emission_utils.cc136 absl::optional<CudaComputeCapability> cuda_compute_capability) { in GetReductionTiling()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
Dmlir_gpu_test_base.cc56 absl::optional<CudaComputeCapability> cuda_compute_capability = in CompileMlirModule() local
58 CudaComputeCapability cuda_compute_capability; in CompileMlirModule() local
Dhlo_to_llvm_ir.cc53 xla::gpu::CudaComputeCapability cuda_compute_capability; in CompileAndPrintLlvmIr() local
/external/tensorflow/tensorflow/core/grappler/
Ddevices.cc55 std::pair<int, int> cuda_compute_capability(cc_major, cc_minor); in GetNumAvailableGPUs() local
/external/tensorflow/tensorflow/stream_executor/
Ddevice_description.cc114 bool DeviceDescription::cuda_compute_capability(int *major, int *minor) const { in cuda_compute_capability() function in stream_executor::DeviceDescription