Home
last modified time | relevance | path

Searched refs:GetCudaComputeCapability (Results 1 – 12 of 12) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
H A Dgemm_rewrite_test.cc61 se::CudaComputeCapability GetCudaComputeCapability() { in GetCudaComputeCapability() function in xla::gpu::__anonefce84000111::GemmRewriteTest
685 if (GetCudaComputeCapability().IsAtLeast(se::CudaComputeCapability::AMPERE)) {
714 if (GetCudaComputeCapability().IsAtLeast(se::CudaComputeCapability::AMPERE)) {
784 if (GetCudaComputeCapability().IsAtLeast(se::CudaComputeCapability::VOLTA)) {
815 if (GetCudaComputeCapability().IsAtLeast(se::CudaComputeCapability::VOLTA)) {
849 if (GetCudaComputeCapability().IsAtLeast(se::CudaComputeCapability::VOLTA)) {
882 if (GetCudaComputeCapability().IsAtLeast(se::CudaComputeCapability::VOLTA)) {
H A Dmlir_gemm_test.cc69 if (stream->GetCudaComputeCapability().IsAtLeast( in TEST_F()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
H A Dcuda_blas.cc624 if (stream->GetCudaComputeCapability().IsAtLeast( in DoBlasInternalImpl()
928 CudaComputeCapability cc = stream->GetCudaComputeCapability(); in DoBlasInternalImpl()
1063 !stream->GetCudaComputeCapability().IsAtLeast(7)) { in DoBlasInternalImpl()
1105 if (stream->GetCudaComputeCapability().IsAtLeast( in DoBlasInternalImpl()
1256 if (stream->GetCudaComputeCapability().IsAtLeast(5)) { in DoBlasInternalImpl()
1427 CudaComputeCapability cc = stream->GetCudaComputeCapability(); in DoBlasInternalImpl()
1463 CudaComputeCapability cc = stream->GetCudaComputeCapability(); in DoBlasInternalImpl()
H A Dcuda_dnn.cc917 if (!TensorOpMathAvailable(stream->GetCudaComputeCapability())) { in IsTensorMathEnabled()
4750 auto cuda_compute_capability = stream->GetCudaComputeCapability(); in GetConvolveRunners()
5192 !stream->GetCudaComputeCapability().IsAtLeast(6, 1)) { in GetFusedConvolveRunners()
5219 auto cuda_compute_capability = stream->GetCudaComputeCapability(); in GetFusedConvolveRunners()
5769 !stream->GetCudaComputeCapability().IsAtLeast(6, 1)) { in DoFusedConvolve()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/
H A Dtf_framework_c_interface.cc269 ctx->op_device_context()->stream()->GetCudaComputeCapability(); in _mlir_ciface_tf_jit_compile()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dconv_grad_input_ops.cc218 stream->GetCudaComputeCapability().IsAtLeast( in operator ()()
H A Dconv_ops_fused_impl.h436 stream->GetCudaComputeCapability().IsAtLeast(
H A Dconv_grad_filter_ops.cc837 stream->GetCudaComputeCapability().IsAtLeast( in operator ()()
H A Dconv_ops.cc890 stream->GetCudaComputeCapability().IsAtLeast( in operator ()()
H A Ddepthwise_conv_op_gpu.h678 ctx->op_device_context()->stream()->GetCudaComputeCapability();
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dgpu_executable.cc299 GpuVersion cc = main_stream->GetCudaComputeCapability(); in CheckCompatibilityWithServiceExecutableRunOptions()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/
H A Dstream.h1465 CudaComputeCapability GetCudaComputeCapability() const { in GetCudaComputeCapability() function