Searched refs:ReadStringFromEnvVar (Results 1 – 15 of 15) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/ |
| H A D | auto_mixed_precision_lists.h | 66 TF_CHECK_OK(ReadStringFromEnvVar(add_env_var, "", &to_add)); in UpdateList()
67 TF_CHECK_OK(ReadStringFromEnvVar(remove_env_var, "", &to_remove)); in UpdateList()
102 ReadStringFromEnvVar("TF_AUTO_MIXED_PRECISION_GRAPH_REWRITE_LEVEL", "", in IsPseudoFastMath()
|
| H A D | auto_mixed_precision.cc | 56 TF_CHECK_OK(ReadStringFromEnvVar( in ShouldSimulateGpu()
1188 TF_RETURN_IF_ERROR(ReadStringFromEnvVar( in PrintDebugLogs()
1375 TF_RETURN_IF_ERROR(ReadStringFromEnvVar( in Optimize()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/util/ |
| H A D | env_var.cc | 78 Status ReadStringFromEnvVar(StringPiece env_var_name, StringPiece default_val, in ReadStringFromEnvVar() function
92 TF_RETURN_IF_ERROR(ReadStringFromEnvVar(env_var_name, default_val, &str_val)); in ReadStringsFromEnvVar()
|
| H A D | env_var.h | 46 Status ReadStringFromEnvVar(StringPiece env_var_name, StringPiece default_val,
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/ |
| H A D | trt_experimental_features.cc | 28 ReadStringFromEnvVar("TF_TRT_EXPERIMENTAL_FEATURES", "", &envvar_str)); in isExperimentalFeatureActivated()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/platform/ |
| H A D | enable_tf2_utils_test.cc | 26 TF_CHECK_OK(ReadStringFromEnvVar("TF2_BEHAVIOR", "0", &tf2_env)); in TEST()
|
| H A D | enable_tf2_utils.cc | 40 TF_CHECK_OK(ReadStringFromEnvVar("TF2_BEHAVIOR", "0", &tf2_env)); in tf2_execution_enabled()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/fallback/ |
| H A D | cost_recorder.cc | 52 TF_RETURN_IF_ERROR(ReadStringFromEnvVar("TF_TFRT_MEASURED_COST_PATH", "", in WriteToFile()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
| H A D | node_file_writer.cc | 76 ReadStringFromEnvVar("TF_NODE_FILE_WRITER_DIRECTORY", "", &node_dir)); in GetNodeFileWriterIfEnabled()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfr/integration/ |
| H A D | tfr_decompose_ctx.cc | 64 TF_RETURN_IF_ERROR(ReadStringFromEnvVar( in Get()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/pluggable_device/ |
| H A D | pluggable_device.cc | 208 TF_RETURN_IF_ERROR(ReadStringFromEnvVar("TF_GPU_THREAD_MODE", "global", in Init()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/segment/ |
| H A D | segment.cc | 737 TF_CHECK_OK(ReadStringFromEnvVar("TF_TRT_SHOW_DETAILED_REPORT", in GenerateNonConversionReport()
874 TF_CHECK_OK(ReadStringFromEnvVar("TF_TRT_OP_DENYLIST", /*default_value=*/"", in SegmentGraph()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| H A D | grpc_state.h | 65 TF_CHECK_OK(ReadStringFromEnvVar("GRPC_FAIL_FAST", "use_caller",
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/platform/cloud/ |
| H A D | curl_http_request.cc | 134 TF_CHECK_OK(ReadStringFromEnvVar("CURL_CA_BUNDLE", "", &value)); in CurlHttpRequest()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| H A D | gpu_device.cc | 512 ReadStringFromEnvVar("TF_GPU_THREAD_MODE", "global", &gpu_thread_mode)); in Init()
|