HomeSort by relevance Sort by last modified time
    Searched refs:cuda_compute_capability (Results 1 - 10 of 10) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/
platform_util.cc 190 if (description.cuda_compute_capability(&major_version, &minor_version)) {
  /external/tensorflow/tensorflow/stream_executor/
device_description.cc 112 bool DeviceDescription::cuda_compute_capability(int *major, int *minor) const { function in class:perftools::gputools::DeviceDescription
device_description.h 156 bool cuda_compute_capability(int *major, int *minor) const;
stream_executor_pimpl.cc 303 GetDeviceDescription().cuda_compute_capability(&cc_major, &cc_minor);
316 GetDeviceDescription().cuda_compute_capability(&cc_major, &cc_minor);
329 GetDeviceDescription().cuda_compute_capability(&cc_major, &cc_minor);
  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
gpu_executable.cc 136 main_stream->parent()->GetDeviceDescription().cuda_compute_capability(
gpu_compiler.cc 601 if (!stream_exec->GetDeviceDescription().cuda_compute_capability(&cc_major,
  /external/tensorflow/tensorflow/core/common_runtime/gpu/
gpu_device.cc     [all...]
  /external/tensorflow/tensorflow/contrib/fused_conv/kernels/
fused_conv2d_bias_activation_op.cc 326 stream->parent()->GetDeviceDescription().cuda_compute_capability(&cc_major,
  /external/tensorflow/tensorflow/stream_executor/cuda/
cuda_blas.cc     [all...]
cuda_dnn.cc     [all...]

Completed in 448 milliseconds