Searched refs:cc_minor_ (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_gpu_executor.cc | 150 return GpuDriver::GetComputeCapability(&cc_major_, &cc_minor_, device_); in Init()
156 if (cc_major_ == 0 && cc_minor_ == 0) { in FindOnDiskForComputeCapability()
161 absl::StrCat(filename, ".cc", cc_major_, cc_minor_, canonical_suffix); in FindOnDiskForComputeCapability()
286 if (cc_major_ == 0 && cc_minor_ == 0) { in GetKernel()
290 const char *ptx = spec.cuda_ptx_in_memory().text(cc_major_, cc_minor_); in GetKernel()
376 if (cc_major_ == 0 && cc_minor_ == 0) { in LoadModule()
1114 absl::StrCat("Compute Capability ", cc_major_, ".", cc_minor_)); in PopulateDeviceDescription()
1121 builder.set_cuda_compute_capability(cc_major_, cc_minor_); in PopulateDeviceDescription()
|
| /external/tensorflow/tensorflow/stream_executor/gpu/ |
| D | gpu_executor.h | 53 cc_minor_(0), in GpuExecutor()
333 int cc_minor_; variable
|