Searched defs:cuda_compute_capability (Results 1 – 11 of 11) 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 | 970 se::CudaComputeCapability cuda_compute_capability, in CompileModuleToLlvmIrImpl()
1524 se::CudaComputeCapability cuda_compute_capability, in CompileModuleToLlvmIr()
|
| D | ir_emission_utils.cc | 133 se::CudaComputeCapability cuda_compute_capability) { in GetReductionTiling()
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/lmhlo_to_gpu/ |
| D | lmhlo_to_gpu_binary.h | 53 stream_executor::CudaComputeCapability cuda_compute_capability; member
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
| D | device_description.cc | 109 CudaComputeCapability DeviceDescription::cuda_compute_capability() const { in cuda_compute_capability() function in stream_executor::DeviceDescription
|
| D | dnn.cc | 111 CudaComputeCapability cuda_compute_capability, in GetConvolveAlgorithms()
192 CudaComputeCapability cuda_compute_capability, in GetConvolveBackwardDataAlgorithms()
198 CudaComputeCapability cuda_compute_capability, in GetConvolveBackwardFilterAlgorithms()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/ |
| D | hlo_to_llvm_ir.cc | 67 tensorflow::se::CudaComputeCapability cuda_compute_capability; in CompileAndPrintLlvmIr() local
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| D | cuda_dnn.cc | 912 CudaComputeCapability cuda_compute_capability) { in TensorOpMathAvailable()
4750 auto cuda_compute_capability = stream->GetCudaComputeCapability(); in GetConvolveRunners() local
5219 auto cuda_compute_capability = stream->GetCudaComputeCapability(); in GetFusedConvolveRunners() local
5275 CudaComputeCapability cuda_compute_capability, in GetConvolveAlgorithms()
5334 CudaComputeCapability cuda_compute_capability, in GetConvolveBackwardDataAlgorithms()
5369 CudaComputeCapability cuda_compute_capability, in GetConvolveBackwardFilterAlgorithms()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 3133 CudaComputeCapability cuda_compute_capability, in GetConvolveAlgorithms()
3610 CudaComputeCapability cuda_compute_capability, in GetConvolveBackwardDataAlgorithms()
3625 CudaComputeCapability cuda_compute_capability, in GetConvolveBackwardFilterAlgorithms()
|