Searched refs:ReadBoolFromEnvVar (Results 1 – 15 of 15) sorted by relevance
| /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 | 2196 TF_CHECK_OK(ReadBoolFromEnvVar("TF_MKL_OPTIMIZE_PRIMITIVE_MEMUSE", true, in IsPrimitiveMemOptEnabled()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | context.cc | 36 bool ReadBoolFromEnvVar(StringPiece env_var_name, bool default_val) { in ReadBoolFromEnvVar() function
38 if (tensorflow::ReadBoolFromEnvVar(env_var_name, default_val, &val).ok()) { in ReadBoolFromEnvVar()
71 pin_small_ops_to_cpu_(ReadBoolFromEnvVar( in EagerContext()
473 ReadBoolFromEnvVar("TF_EAGER_REMOTE_USE_SEND_TENSOR_RPC", false); in InitializeRemote()
|
| /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 | 302 ReadBoolFromEnvVar("TF_SYNC_ON_FINISH", true, &sync_on_finish_); in DirectSession()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | maxpooling_op.cc | 374 TF_CHECK_OK(ReadBoolFromEnvVar("TF_ENABLE_MAXPOOL_NANPROP", false, in MaxPoolingGradOp()
917 TF_CHECK_OK(ReadBoolFromEnvVar("TF_ENABLE_MAXPOOL_NANPROP", false, in MaxPoolingWithArgmaxOp()
1142 TF_CHECK_OK(ReadBoolFromEnvVar("TF_ENABLE_MAXPOOL_NANPROP", false, in MaxPoolingNoMaskOp()
1234 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 | cudnn_rnn_ops.cc | 919 OP_REQUIRES_OK(context, ReadBoolFromEnvVar("TF_CUDNN_RESET_RND_GEN_STATE", in CudnnRNNKernelCommon()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | graph_mgr.cc | 61 ReadBoolFromEnvVar("TF_SYNC_ON_FINISH", true, &sync_on_finish_); in GraphMgr()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 593 tensorflow::ReadBoolFromEnvVar("TF_DISABLE_CUDNN_TENSOR_OP_MATH", in TensorOpMathEnabled()
606 tensorflow::ReadBoolFromEnvVar("TF_DISABLE_CUDNN_RNN_TENSOR_OP_MATH", in RnnTensorOpMathEnabled()
625 TF_CHECK_OK(tensorflow::ReadBoolFromEnvVar( in BatchnormSpatialPersistentEnabled()
637 TF_CHECK_OK(tensorflow::ReadBoolFromEnvVar("TF_CUDNN_DETERMINISTIC", in RequireDeterminism()
|
| D | cuda_blas.cc | 111 tensorflow::ReadBoolFromEnvVar("TF_DISABLE_CUBLAS_TENSOR_OP_MATH", in TensorOpMathEnabled()
|