Searched defs:cuda_compute_capability (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_context.h | 42 se::CudaComputeCapability cuda_compute_capability, in IrEmitterContext() 64 se::CudaComputeCapability cuda_compute_capability() const { in cuda_compute_capability() function
|
| D | tree_reduction_rewriter.h | 79 se::CudaComputeCapability cuda_compute_capability) in GpuTreeReductionRewriter()
|
| D | tree_reduction_rewriter.cc | 49 se::CudaComputeCapability cuda_compute_capability) in ReductionRewriterVisitor()
|
| D | gpu_compiler.cc | 745 se::CudaComputeCapability cuda_compute_capability, in CompileModuleToLlvmIrImpl() 1177 se::CudaComputeCapability cuda_compute_capability, int pointer_size) { in CompileModuleToLlvmIr()
|
| D | ir_emission_utils.cc | 152 se::CudaComputeCapability cuda_compute_capability) { in GetReductionTiling()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/ |
| D | hlo_to_llvm_ir.cc | 66 tensorflow::se::CudaComputeCapability cuda_compute_capability; in CompileAndPrintLlvmIr() local
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | device_description.cc | 111 CudaComputeCapability DeviceDescription::cuda_compute_capability() const { in cuda_compute_capability() function in stream_executor::DeviceDescription
|
| D | dnn.cc | 52 CudaComputeCapability cuda_compute_capability, in GetConvolveAlgorithms() 86 CudaComputeCapability cuda_compute_capability, in GetConvolveBackwardDataAlgorithms() 92 CudaComputeCapability cuda_compute_capability, in GetConvolveBackwardFilterAlgorithms()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 837 CudaComputeCapability cuda_compute_capability) { in TensorOpMathAvailable() 4406 CudaComputeCapability cuda_compute_capability, in GetConvolveAlgorithms() 4473 CudaComputeCapability cuda_compute_capability, in GetConvolveBackwardDataAlgorithms() 4513 CudaComputeCapability cuda_compute_capability, in GetConvolveBackwardFilterAlgorithms()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 3078 CudaComputeCapability cuda_compute_capability, in GetConvolveAlgorithms() 3489 CudaComputeCapability cuda_compute_capability, in GetConvolveBackwardDataAlgorithms() 3504 CudaComputeCapability cuda_compute_capability, in GetConvolveBackwardFilterAlgorithms()
|