Searched refs:cuda_compute_capability (Results 1 – 15 of 15) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/ |
| D | device_description.cc | 111 bool DeviceDescription::cuda_compute_capability(int *major, int *minor) const { in cuda_compute_capability() function in stream_executor::DeviceDescription
|
| D | device_description.h | 138 bool cuda_compute_capability(int *major, int *minor) const;
|
| D | stream_executor_pimpl.cc | 331 GetDeviceDescription().cuda_compute_capability(&cc_major, &cc_minor); in GetConvolveAlgorithms()
353 GetDeviceDescription().cuda_compute_capability(&cc_major, &cc_minor); in GetConvolveBackwardDataAlgorithms()
366 GetDeviceDescription().cuda_compute_capability(&cc_major, &cc_minor); in GetConvolveBackwardFilterAlgorithms()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | stream_executor_util.cc | 32 CHECK(stream_executor.GetDeviceDescription().cuda_compute_capability(&major, in IsVoltaOrLater()
|
| D | gpu_executable.cc | 74 executor->GetDeviceDescription().cuda_compute_capability( in ExecuteThunks()
|
| D | cudnn_conv_algorithm_picker.cc | 118 stream_executor->GetDeviceDescription().cuda_compute_capability(&cc_major, in GetComputeCapability()
|
| D | nvptx_compiler.cc | 738 if (!stream_exec->GetDeviceDescription().cuda_compute_capability(&cc_major, in RunBackend()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | gpu_utils.cc | 49 stream_executor->GetDeviceDescription().cuda_compute_capability(&cc_major, in GetComputeCapability()
|
| D | depthwise_conv_op_gpu.h | 669 .cuda_compute_capability(&major, &minor);
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | platform_util.cc | 193 if (description.cuda_compute_capability(&major_version, &minor_version)) { in IsDeviceSupported()
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_device.cc | 905 if (!se->GetDeviceDescription().cuda_compute_capability(&cc_major, in SingleVirtualDeviceMemoryLimit()
1166 if (!desc.cuda_compute_capability(&cc_major, &cc_minor)) { in GetShortDeviceDescription()
1546 if (!description.cuda_compute_capability(&cc_major, &cc_minor)) { in GetValidDeviceIds()
1617 if (!desc.cuda_compute_capability(&device_capability.major_part, in GetValidDeviceIds()
|
| D | gpu_device_test.cc | 49 if (!se->GetDeviceDescription().cuda_compute_capability(cc_major, cc_minor)) { in GetComputeCapability()
|
| /external/tensorflow/tensorflow/contrib/fused_conv/kernels/ |
| D | fused_conv2d_bias_activation_op.cc | 305 stream_executor->GetDeviceDescription().cuda_compute_capability(&cc_major, in GetComputeCapability()
461 stream->parent()->GetDeviceDescription().cuda_compute_capability(&cc_major, in launch()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_blas.cc | 1599 stream->parent()->GetDeviceDescription().cuda_compute_capability(&cc_major, in DoBlasGemm()
1896 if (stream->parent()->GetDeviceDescription().cuda_compute_capability( in DoBlasGemmWithAlgorithmImpl()
2214 if (stream->parent()->GetDeviceDescription().cuda_compute_capability( in DoBlasGemmBatchedInternal()
2367 if (stream->parent()->GetDeviceDescription().cuda_compute_capability( in DoBlasGemmStridedBatched()
|
| D | cuda_dnn.cc | 3430 stream->parent()->GetDeviceDescription().cuda_compute_capability(&cc_major, in DoFusedConvolve()
|