Home
last modified time | relevance | path

Searched refs:ReadInt64FromEnvVar (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Dgrpc_worker_cache.cc149 Status status = ReadInt64FromEnvVar("TF_GRPC_WORKER_CACHE_QUEUES", 64, in CreateGrpcWorkerEnv()
155 status = ReadInt64FromEnvVar("TF_GRPC_WORKER_CACHE_THREADS", num_cpus, in CreateGrpcWorkerEnv()
Dgrpc_remote_worker.cc306 TF_CHECK_OK(ReadInt64FromEnvVar("GRPC_MAX_RETRIES", 0, &max_retries)); in MaxRetries()
/external/tensorflow/tensorflow/core/util/
Denv_var.h36 Status ReadInt64FromEnvVar(StringPiece env_var_name, int64 default_val,
Duse_cudnn.cc61 Status status = ReadInt64FromEnvVar(#flag_name, default_value, &value); \
Denv_var.cc48 Status ReadInt64FromEnvVar(StringPiece env_var_name, int64 default_val, in ReadInt64FromEnvVar() function
/external/tensorflow/tensorflow/core/common_runtime/
Dprocess_state.cc95 Status status = ReadInt64FromEnvVar("TF_CPU_BFC_MEM_LIMIT_IN_MB", in GetCPUAllocator()
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_process_state.cc304 Status status = ReadInt64FromEnvVar("TF_GPU_HOST_MEM_LIMIT_IN_MB", in GetGpuHostAllocator()
Dgpu_device.cc500 ReadInt64FromEnvVar("TF_GPU_THREAD_COUNT", 2, &gpu_thread_count)); in Init()
/external/tensorflow/tensorflow/core/kernels/
Dfft_ops.cc407 Status status = ReadInt64FromEnvVar(envvar_in_mb, default_value_in_bytes, in GetCufftWorkspaceLimit()
/external/tensorflow/tensorflow/stream_executor/
Dstream_executor_pimpl.cc134 SE_CHECK_OK(tensorflow::ReadInt64FromEnvVar("TF_PER_DEVICE_MEMORY_LIMIT_MB", in GetMemoryLimitBytes()
/external/tensorflow/tensorflow/core/util/tensor_bundle/
Dtensor_bundle.cc766 ReadInt64FromEnvVar("TF_TABLE_INDEX_CACHE_SIZE_IN_MB", 0, &cache_size); in BundleReader()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc1472 TF_CHECK_OK(tensorflow::ReadInt64FromEnvVar("TF_TRT_FIXED_ALGORITHM_ID", in BuildCudaEngine()