Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dir_emitter_context.h42 se::CudaComputeCapability cuda_compute_capability, in IrEmitterContext()
64 se::CudaComputeCapability cuda_compute_capability() const { in cuda_compute_capability() function
H A Dtree_reduction_rewriter.h79 se::CudaComputeCapability cuda_compute_capability) in GpuTreeReductionRewriter()
H A Dtree_reduction_rewriter.cc49 se::CudaComputeCapability cuda_compute_capability) in ReductionRewriterVisitor()
H A Dgpu_compiler.cc970 se::CudaComputeCapability cuda_compute_capability, in CompileModuleToLlvmIrImpl()
1524 se::CudaComputeCapability cuda_compute_capability, in CompileModuleToLlvmIr()
H A Dir_emission_utils.cc133 se::CudaComputeCapability cuda_compute_capability) { in GetReductionTiling()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/
H A Ddnn.cc111 CudaComputeCapability cuda_compute_capability, in GetConvolveAlgorithms()
192 CudaComputeCapability cuda_compute_capability, in GetConvolveBackwardDataAlgorithms()
198 CudaComputeCapability cuda_compute_capability, in GetConvolveBackwardFilterAlgorithms()
H A Ddevice_description.cc109 CudaComputeCapability DeviceDescription::cuda_compute_capability() const { in cuda_compute_capability() function in stream_executor::DeviceDescription
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/lmhlo_to_gpu/
H A Dlmhlo_to_gpu_binary.h53 stream_executor::CudaComputeCapability cuda_compute_capability; member
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
H A Dhlo_to_llvm_ir.cc67 tensorflow::se::CudaComputeCapability cuda_compute_capability; in CompileAndPrintLlvmIr() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
H A Dcuda_dnn.cc912 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()
/aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/
H A Drocm_dnn.cc3133 CudaComputeCapability cuda_compute_capability, in GetConvolveAlgorithms()
3610 CudaComputeCapability cuda_compute_capability, in GetConvolveBackwardDataAlgorithms()
3625 CudaComputeCapability cuda_compute_capability, in GetConvolveBackwardFilterAlgorithms()