Searched refs:GetComputeCapability (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | gpu_utils.cc | 45 tensorflow::ComputeCapability GetComputeCapability( in GetComputeCapability() function
73 *log.mutable_compute_capability() = GetComputeCapability(stream_exec); in LogConvAutotuneResults()
103 *log.mutable_compute_capability() = GetComputeCapability(stream_exec); in LogFusedConvAutotuneResults()
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_device_test.cc | 44 Status GetComputeCapability(PlatformGpuId gpu_id, int* cc_major, in GetComputeCapability() function
226 TF_ASSERT_OK(GetComputeCapability(PlatformGpuId(0), &cc_major, &cc_minor)); in TEST_F()
250 TF_ASSERT_OK(GetComputeCapability(kPlatformGpuId, &cc_major, &cc_minor)); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cudnn_conv_algorithm_picker.cc | 114 tensorflow::ComputeCapability GetComputeCapability( in GetComputeCapability() function
324 *log.mutable_compute_capability() = GetComputeCapability(stream_exec_); in PickBestAlgorithm()
|
| /external/tensorflow/tensorflow/stream_executor/gpu/ |
| D | gpu_driver.h | 398 static port::Status GetComputeCapability(int* cc_major, int* cc_minor,
|
| /external/tensorflow/tensorflow/contrib/fused_conv/kernels/ |
| D | fused_conv2d_bias_activation_op.cc | 301 tensorflow::ComputeCapability GetComputeCapability( in GetComputeCapability() function
336 internal::GetComputeCapability(stream_exec); in LogFusedConvAutotuneResults()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_gpu_executor.cc | 150 return GpuDriver::GetComputeCapability(&cc_major_, &cc_minor_, device_); in Init()
|
| D | cuda_driver.cc | 1350 /* static */ port::Status GpuDriver::GetComputeCapability(int* cc_major, in GetComputeCapability() function in stream_executor::gpu::GpuDriver
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_driver.cc | 1055 /* static */ port::Status GpuDriver::GetComputeCapability(int* cc_major, in GetComputeCapability() function in stream_executor::gpu::GpuDriver
|