Home
last modified time | relevance | path

Searched refs:gpu_version_ (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_executable.cc63 gpu_version_(params.gpu_version), in GpuExecutable()
105 absl::get<std::pair<int, std::string>>(gpu_version_).first; in CheckCompatibilityWithServiceExecutableRunOptions()
115 TF_RET_CHECK(nvidia_compute_compatibility == gpu_version_) in CheckCompatibilityWithServiceExecutableRunOptions()
117 << absl::get<std::pair<int, int>>(gpu_version_).first << ", " in CheckCompatibilityWithServiceExecutableRunOptions()
118 << absl::get<std::pair<int, int>>(gpu_version_).second << "}, but was {" in CheckCompatibilityWithServiceExecutableRunOptions()
Dgpu_executable.h190 GpuVersion gpu_version_; variable