| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_device_test.cc | 44 Status GetComputeCapability(PlatformGpuId gpu_id, int* cc_major, in GetComputeCapability()
225 int cc_major, cc_minor; in TEST_F() local
249 int cc_major, cc_minor; in TEST_F() local
|
| D | gpu_device.cc | 904 int cc_major = 0, cc_minor = 0; in SingleVirtualDeviceMemoryLimit() local
1164 int cc_major; in GetShortDeviceDescription() local
1544 int cc_major; in GetValidDeviceIds() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | nvptx_compiler.h | 120 int cc_major; member
|
| D | nvptx_compiler.cc | 540 const string& ptx, int cc_major, int cc_minor, in CompilePtx()
737 int cc_major, cc_minor; in RunBackend() local
804 const string& ptx, int cc_major, int cc_minor, in CompilePtxOrGetCachedResult()
|
| D | cudnn_conv_algorithm_picker.cc | 117 int cc_major, cc_minor; in GetComputeCapability() local
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | dnn.cc | 30 bool with_winograd_nonfused, int cc_major, int cc_minor, in GetConvolveAlgorithms()
40 bool with_winograd_nonfused, int cc_major, int cc_minor, in GetConvolveBackwardDataAlgorithms()
46 bool with_winograd_nonfused, int cc_major, int cc_minor, in GetConvolveBackwardFilterAlgorithms()
|
| D | stream_executor_pimpl.cc | 330 int cc_major, cc_minor; in GetConvolveAlgorithms() local
352 int cc_major, cc_minor; in GetConvolveBackwardDataAlgorithms() local
365 int cc_major, cc_minor; in GetConvolveBackwardFilterAlgorithms() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | gpu_utils.cc | 48 int cc_major, cc_minor; in GetComputeCapability() local
|
| /external/tensorflow/tensorflow/contrib/fused_conv/kernels/ |
| D | fused_conv2d_bias_activation_op.cc | 304 int cc_major, cc_minor; in GetComputeCapability() local
460 int cc_major, cc_minor; in launch() local
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 3038 inline bool TensorOpMathAvailable(int cc_major) { in TensorOpMathAvailable()
3043 bool with_winograd_nonfused, int cc_major, int cc_minor, in GetConvolveAlgorithms()
3104 bool with_winograd_nonfused, int cc_major, int cc_minor, in GetConvolveBackwardDataAlgorithms()
3139 bool with_winograd_nonfused, int cc_major, int cc_minor, in GetConvolveBackwardFilterAlgorithms()
3429 int cc_major, cc_minor; in DoFusedConvolve() local
|
| D | cuda_blas.cc | 1598 int cc_major, cc_minor; in DoBlasGemm() local
1872 static bool TensorOpsAvailable(int cc_major) { in TensorOpsAvailable()
1895 int cc_major, cc_minor; in DoBlasGemmWithAlgorithmImpl() local
2213 int cc_major, cc_minor; in DoBlasGemmBatchedInternal() local
2366 int cc_major, cc_minor; in DoBlasGemmStridedBatched() local
|
| D | cuda_driver.cc | 1350 /* static */ port::Status GpuDriver::GetComputeCapability(int* cc_major, in GetComputeCapability()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_driver.cc | 1055 /* static */ port::Status GpuDriver::GetComputeCapability(int* cc_major, in GetComputeCapability()
|
| D | rocm_dnn.cc | 2949 bool with_winograd_nonfused, int cc_major, int cc_minor, in GetConvolveAlgorithms()
2970 bool with_winograd_nonfused, int cc_major, int cc_minor, in GetConvolveBackwardDataAlgorithms()
2985 bool with_winograd_nonfused, int cc_major, int cc_minor, in GetConvolveBackwardFilterAlgorithms()
|