Add new env-var TF_FP16_CONV_MODE.

The env-var is used to decide how to do internal computation for cudnn convolution when input data type is float16.
For ACCURATE mode, we always use float32 as the internal compute type;
For FAST mode, we include both float16 and float32 internal compute type into auto-tune to pick whichever runs faster.

PiperOrigin-RevId: 174495814

1 files changed, 11 insertions, 0 deletions

diff --git a/tensorflow/core/util/env_var.cc b/tensorflow/core/util/env_var.cc
index d4e89b966e..c844850179 100644
--- a/tensorflow/core/util/env_var.cc
+++ b/tensorflow/core/util/env_var.cc
@@ -60,4 +60,15 @@ Status ReadInt64FromEnvVar(StringPiece env_var_name, int64 default_val,
       tf_env_var_val, ". Use the default value: ", default_val));
 }
 
+Status ReadStringFromEnvVar(StringPiece env_var_name, StringPiece default_val,
+                            string* value) {
+  const char* tf_env_var_val = getenv(env_var_name.ToString().c_str());
+  if (tf_env_var_val != nullptr) {
+    *value = tf_env_var_val;
+  } else {
+    *value = default_val.ToString();
+  }
+  return Status::OK();
+}
+
 }  // namespace tensorflow