| /external/tensorflow/tensorflow/core/util/ |
| matmul_autotune.cc | 26 ReadBoolFromEnvVar("TF_MATMUL_AUTOTUNE_ENABLE", false, &value);
44 ReadBoolFromEnvVar("TF_FP16_MATMUL_USE_FP32_COMPUTE", true, &value);
|
| env_var.h | 29 Status ReadBoolFromEnvVar(StringPiece env_var_name, bool default_val,
|
| use_cudnn.cc | 28 Status status = ReadBoolFromEnvVar(#flag_name, default_value, &value); \
|
| env_var.cc | 28 Status ReadBoolFromEnvVar(StringPiece env_var_name, bool default_val,
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| process_state.cc | 188 Status status = ReadBoolFromEnvVar("TF_CPU_ALLOCATOR_USE_BFC", false,
|
| /external/tensorflow/tensorflow/core/kernels/ |
| maxpooling_op.cc | 360 TF_CHECK_OK(ReadBoolFromEnvVar("TF_ENABLE_MAXPOOL_NANPROP", false,
[all...] |
| deep_conv2d.cc | 81 static bool ReadBoolFromEnvVar(const char* env_var_name, bool default_val) {
109 if (!ReadBoolFromEnvVar("TF_USE_DEEP_CONV2D", false)) {
[all...] |
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| graph_mgr.cc | 57 ReadBoolFromEnvVar("TF_SYNC_ON_FINISH", true, &sync_on_finish_);
|
| /external/tensorflow/tensorflow/contrib/cudnn_rnn/kernels/ |
| cudnn_rnn_ops.cc | 503 OP_REQUIRES_OK(context, ReadBoolFromEnvVar("TF_CUDNN_RESET_RND_GEN_STATE",
[all...] |
| /external/tensorflow/tensorflow/core/common_runtime/ |
| direct_session.cc | 281 ReadBoolFromEnvVar("TF_SYNC_ON_FINISH", true, &sync_on_finish_);
[all...] |
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| cuda_dnn.cc | 582 tensorflow::ReadBoolFromEnvVar("TF_DISABLE_CUDNN_TENSOR_OP_MATH",
600 TF_CHECK_OK(tensorflow::ReadBoolFromEnvVar(
[all...] |
| cuda_blas.cc | 313 tensorflow::ReadBoolFromEnvVar("TF_DISABLE_CUBLAS_TENSOR_OP_MATH",
[all...] |
