Home
last modified time | relevance | path

Searched refs:version_or (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dgpu_utils.cc102 se::port::StatusOr<se::dnn::VersionInfo> version_or = dnn->GetVersion(); in GetCudnnVersion() local
103 if (version_or.ok()) { in GetCudnnVersion()
104 const auto& version = version_or.ValueOrDie(); in GetCudnnVersion()
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf_with_tf2xla.cc387 tensorflow::StatusOr<int64_t> version_or = in PrepareParams() local
390 if (!version_or.ok()) { in PrepareParams()
391 return emitError(op_->getLoc()) << version_or.status().ToString(); in PrepareParams()
398 version_or.ValueOrDie(), flib_def_.get(), tensorflow::OptimizerOptions()); in PrepareParams()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_conv_algorithm_picker.cc167 StatusOr<se::dnn::VersionInfo> version_or = dnn->GetVersion(); in GetCudnnVersion() local
168 if (version_or.ok()) { in GetCudnnVersion()
169 const auto& version = version_or.ValueOrDie(); in GetCudnnVersion()