Function GetComputeCapability

tensorflow/core/kernels/gpu_utils.cc:122–131

}

tensorflow::ComputeCapability GetComputeCapability(
    se::StreamExecutor* stream_executor) {
  tensorflow::ComputeCapability cc;
  int cc_major, cc_minor;
  stream_executor->GetDeviceDescription().cuda_compute_capability(&cc_major,
                                                                  &cc_minor);
  cc.set_major(cc_major);
  cc.set_minor(cc_minor);
  return cc;
}

template<typename TResult, typename TLog>
void LogConvAutotuneResultsImpl(se::dnn::ConvolutionKind kind,