| 26 | namespace tensorflow { |
| 27 | |
| 28 | Status ReadBoolFromEnvVar(StringPiece env_var_name, bool default_val, |
| 29 | bool* value) { |
| 30 | *value = default_val; |
| 31 | const char* tf_env_var_val = getenv(string(env_var_name).c_str()); |
| 32 | if (tf_env_var_val == nullptr) { |
| 33 | return Status::OK(); |
| 34 | } |
| 35 | string str_value = absl::AsciiStrToLower(tf_env_var_val); |
| 36 | if (str_value == "0" || str_value == "false") { |
| 37 | *value = false; |
| 38 | return Status::OK(); |
| 39 | } else if (str_value == "1" || str_value == "true") { |
| 40 | *value = true; |
| 41 | return Status::OK(); |
| 42 | } |
| 43 | return errors::InvalidArgument(strings::StrCat( |
| 44 | "Failed to parse the env-var ${", env_var_name, "} into bool: ", |
| 45 | tf_env_var_val, ". Use the default value: ", default_val)); |
| 46 | } |
| 47 | |
| 48 | Status ReadInt64FromEnvVar(StringPiece env_var_name, int64 default_val, |
| 49 | int64* value) { |
no test coverage detected