| /external/tensorflow/tensorflow/core/util/ |
| D | matmul_autotune.cc | 26 ReadBoolFromEnvVar("TF_MATMUL_AUTOTUNE_ENABLE", false, &value); in MatmulAutotuneEnable()
44 ReadBoolFromEnvVar("TF_FP16_MATMUL_USE_FP32_COMPUTE", true, &value); in MatmulDoFP32ComputationFP16Input()
|
| D | env_var.h | 30 Status ReadBoolFromEnvVar(StringPiece env_var_name, bool default_val,
|
| D | use_cudnn.cc | 28 Status status = ReadBoolFromEnvVar(#flag_name, default_value, &value); \
|
| D | env_var.cc | 28 Status ReadBoolFromEnvVar(StringPiece env_var_name, bool default_val, in ReadBoolFromEnvVar() function
|
| D | mkl_util.h | 1976 TF_CHECK_OK(ReadBoolFromEnvVar("TF_MKL_OPTIMIZE_PRIMITIVE_MEMUSE", true,
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | gpu_utils.cc | 232 TF_CHECK_OK(tensorflow::ReadBoolFromEnvVar("TF_DETERMINISTIC_OPS", in RequireCudnnDeterminism()
236 TF_CHECK_OK(tensorflow::ReadBoolFromEnvVar("TF_CUDNN_DETERMINISTIC", in RequireCudnnDeterminism()
|
| D | maxpooling_op.cc | 376 TF_CHECK_OK(ReadBoolFromEnvVar("TF_ENABLE_MAXPOOL_NANPROP", false, in MaxPoolingGradOp()
919 TF_CHECK_OK(ReadBoolFromEnvVar("TF_ENABLE_MAXPOOL_NANPROP", false, in MaxPoolingWithArgmaxOp()
1144 TF_CHECK_OK(ReadBoolFromEnvVar("TF_ENABLE_MAXPOOL_NANPROP", false, in MaxPoolingNoMaskOp()
1236 TF_CHECK_OK(ReadBoolFromEnvVar("TF_ENABLE_MAXPOOL_NANPROP", false, in MaxPoolingNoMaskV2Op()
|
| D | deep_conv2d.cc | 81 static bool ReadBoolFromEnvVar(const char* env_var_name, bool default_val) { in ReadBoolFromEnvVar() function
109 if (!ReadBoolFromEnvVar("TF_USE_DEEP_CONV2D", false)) { in CanUseDeepConv2D()
|
| D | fused_batch_norm_op.cc | 547 TF_CHECK_OK(tensorflow::ReadBoolFromEnvVar( in BatchnormSpatialPersistentEnabled()
|
| D | cudnn_rnn_ops.cc | 989 OP_REQUIRES_OK(context, ReadBoolFromEnvVar("TF_CUDNN_RESET_RND_GEN_STATE", in CudnnRNNKernelCommon()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | process_state.cc | 72 Status status = ReadBoolFromEnvVar( in GetCPUAllocator()
|
| D | local_device.cc | 39 auto status = ReadBoolFromEnvVar("TF_OVERRIDE_GLOBAL_THREADPOOL", in OverrideGlobalThreadPoolFromEnvironment()
|
| D | direct_session.cc | 342 ReadBoolFromEnvVar("TF_SYNC_ON_FINISH", true, &sync_on_finish_); in DirectSession()
|
| /external/tensorflow/tensorflow/core/profiler/internal/cpu/ |
| D | host_tracer.cc | 163 TF_CHECK_OK(ReadBoolFromEnvVar("TF_ENABLE_OSS_CPU_PROFILER", true, &enable)); in __anoneef0a5df0202()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | context.cc | 58 bool ReadBoolFromEnvVar(StringPiece env_var_name, bool default_val) { in ReadBoolFromEnvVar() function
60 if (tensorflow::ReadBoolFromEnvVar(env_var_name, default_val, &val).ok()) { in ReadBoolFromEnvVar()
96 pin_small_ops_to_cpu_(ReadBoolFromEnvVar( in EagerContext()
1032 ReadBoolFromEnvVar("TF_EAGER_REMOTE_USE_SEND_TENSOR_RPC", true); in SetMasterContextState()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | grpc_state.h | 60 TF_CHECK_OK(ReadBoolFromEnvVar("GRPC_FAIL_FAST", fail_fast, &x));
62 TF_CHECK_OK(ReadBoolFromEnvVar("GRPC_FAIL_FAST", false, &x));
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/eager/ |
| D | grpc_eager_client.cc | 60 TF_CHECK_OK(ReadBoolFromEnvVar("TF_ENABLE_EAGER_CLIENT_STREAMING_ENQUEUE", in EnableStreaming()
|
| /external/tensorflow/tensorflow/core/profiler/internal/gpu/ |
| D | device_tracer.cc | 507 ReadBoolFromEnvVar("TF_GPU_CUPTI_USE_ACTIVITY_API", true, in DoStart()
513 ReadBoolFromEnvVar("TF_GPU_CUPTI_FORCE_CONCURRENT_KERNEL", false, in DoStart()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_conv_algorithm_picker.cc | 353 TF_CHECK_OK(tensorflow::ReadBoolFromEnvVar("TF_DETERMINISTIC_OPS", in RequireCudnnDeterminism()
357 TF_CHECK_OK(tensorflow::ReadBoolFromEnvVar("TF_CUDNN_DETERMINISTIC", in RequireCudnnDeterminism()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | graph_mgr.cc | 63 ReadBoolFromEnvVar("TF_SYNC_ON_FINISH", true, &sync_on_finish_); in GraphMgr()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 611 tensorflow::ReadBoolFromEnvVar("TF_DISABLE_CUDNN_TENSOR_OP_MATH", in TensorOpMathEnabled()
624 tensorflow::ReadBoolFromEnvVar("TF_DISABLE_CUDNN_RNN_TENSOR_OP_MATH", in RnnTensorOpMathEnabled()
643 TF_CHECK_OK(tensorflow::ReadBoolFromEnvVar( in BatchnormSpatialPersistentEnabled()
668 TF_CHECK_OK(tensorflow::ReadBoolFromEnvVar("TF_DETERMINISTIC_OPS", in RequireCudnnDeterminism()
672 TF_CHECK_OK(tensorflow::ReadBoolFromEnvVar("TF_CUDNN_DETERMINISTIC", in RequireCudnnDeterminism()
|
| D | cuda_blas.cc | 111 tensorflow::ReadBoolFromEnvVar("TF_DISABLE_CUBLAS_TENSOR_OP_MATH", in TensorOpMathEnabled()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | remapper.cc | 715 TF_CHECK_OK(tensorflow::ReadBoolFromEnvVar( in BatchnormSpatialPersistentEnabled()
|
| D | auto_mixed_precision.cc | 1184 TF_CHECK_OK(ReadBoolFromEnvVar( in ShouldIgnorePerformance()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 3007 tensorflow::ReadBoolFromEnvVar("TF_ROCM_MIMIC_FIND_MODE", false, in GetMIOpenConvolveAlgorithms()
|