| /external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/ |
| D | reduction_vectorization_test.cc | 56 int cc_major = 0, cc_minor = 0; in TEST_F() local
104 int cc_major = 0, cc_minor = 0; in TEST_F() local
152 int cc_major = 0, cc_minor = 0; in TEST_F() local
236 int cc_major = 0, cc_minor = 0; in TEST_F() local
341 int cc_major = 0, cc_minor = 0; in TEST_F() local
|
| D | gpu_atomic_test.cc | 114 int cc_major = 0, cc_minor = 0; in TEST_F() local
|
| /external/tensorflow/tensorflow/stream_executor/gpu/ |
| D | asm_compiler.cc | 146 int cc_major; in CompileGpuAsm() local
180 static void LogPtxasTooOld(const std::string& ptxas_path, int cc_major, in LogPtxasTooOld()
198 port::StatusOr<std::vector<uint8>> CompileGpuAsm(int cc_major, int cc_minor, in CompileGpuAsm()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_device_info.h | 27 int cc_major; member
|
| D | nvptx_compiler.cc | 288 int cc_major, cc_minor; in GetGpuVersion() local
342 se::StreamExecutor* stream_exec, const string& ptx, int cc_major, in CompileGpuAsmOrGetCachedResult()
|
| D | nvptx_compiler.h | 101 int cc_major; member
|
| D | gpu_conv_algorithm_picker.cc | 181 int cc_major, cc_minor; in GetComputeCapability() local
|
| D | ir_emission_utils.cc | 147 int cc_major = 0; in GetReductionTiling() local
|
| D | ir_emitter_unnested.cc | 5347 int cc_major = 0; in ComputeReductionCodegenInfo() local
|
| /external/tensorflow/tensorflow/core/grappler/ |
| D | devices.cc | 52 int cc_major = 0; in GetNumAvailableGPUs() local
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | dnn.cc | 45 bool with_winograd_nonfused, int cc_major, int cc_minor, in GetConvolveAlgorithms()
69 bool with_winograd_nonfused, int cc_major, int cc_minor, in GetConvolveBackwardDataAlgorithms()
75 bool with_winograd_nonfused, int cc_major, int cc_minor, in GetConvolveBackwardFilterAlgorithms()
|
| D | stream_executor_pimpl.cc | 271 int cc_major, cc_minor; in GetConvolveAlgorithms() local
312 int cc_major, cc_minor; in GetConvolveBackwardDataAlgorithms() local
325 int cc_major, cc_minor; in GetConvolveBackwardFilterAlgorithms() local
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_device_test.cc | 43 Status GetComputeCapability(PlatformGpuId gpu_id, int* cc_major, in GetComputeCapability()
352 int cc_major, cc_minor; in TEST_F() local
376 int cc_major, cc_minor; in TEST_F() local
|
| D | gpu_device.cc | 910 int64 MinSystemMemory(int64 available_memory, int cc_major) { in MinSystemMemory()
971 int cc_major = 0, cc_minor = 0; in SingleVirtualDeviceMemoryLimit() local
1135 int cc_major, cc_minor; in GetDeviceDetails() local
1382 int cc_major; in GetShortDeviceDescription() local
1763 int cc_major; in GetValidDeviceIds() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/ |
| D | gpu_kernel_to_blob_pass.cc | 184 uint32_t cc_major = arch / 10; in GetGpuBinaryBlob() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | gpu_utils.cc | 116 int cc_major, cc_minor; in GetComputeCapability() local
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 672 int cc_major, cc_minor; in GetCcMajorMinor() local
755 static bool TensorOpMathAvailable(int cc_major) { return cc_major >= 7; } in TensorOpMathAvailable()
758 int cc_major, cc_minor; in IsTensorMathEnabled() local
3295 bool with_winograd_nonfused, int cc_major, int cc_minor, in GetConvolveAlgorithms()
3365 bool with_winograd_nonfused, int cc_major, int cc_minor, in GetConvolveBackwardDataAlgorithms()
3404 bool with_winograd_nonfused, int cc_major, int cc_minor, in GetConvolveBackwardFilterAlgorithms()
3836 int cc_major, cc_minor; in DoFusedConvolve() local
3867 int cc_major, cc_minor; in DoFusedConvolve() local
|
| D | cuda_gpu_executor.cc | 969 int cc_major; in CreateDeviceDescription() local
|
| D | cuda_driver.cc | 1355 /* static */ port::Status GpuDriver::GetComputeCapability(int* cc_major, in GetComputeCapability()
|
| D | cuda_blas.cc | 1832 int cc_major, cc_minor; in DoBlasInternalImpl() local
2075 int cc_major, cc_minor; in DoBlasInternalImpl() local
2433 int cc_major, cc_minor; in DoBlasInternalImpl() local
2606 int cc_major, cc_minor; in DoBlasInternalImpl() local
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_driver.cc | 992 /* static */ port::Status GpuDriver::GetComputeCapability(int* cc_major, in GetComputeCapability()
|
| D | rocm_dnn.cc | 3084 bool with_winograd_nonfused, int cc_major, int cc_minor, in GetConvolveAlgorithms()
3495 bool with_winograd_nonfused, int cc_major, int cc_minor, in GetConvolveBackwardDataAlgorithms()
3510 bool with_winograd_nonfused, int cc_major, int cc_minor, in GetConvolveBackwardFilterAlgorithms()
|