Searched refs:ReadInt64FromEnvVar (Results 1 – 17 of 17) sorted by relevance
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | grpc_worker_cache.cc | 169 Status status = ReadInt64FromEnvVar("TF_GRPC_WORKER_CACHE_QUEUES", 64, in CreateGrpcWorkerEnv()
175 status = ReadInt64FromEnvVar("TF_GRPC_WORKER_CACHE_THREADS", num_cpus, in CreateGrpcWorkerEnv()
|
| D | grpc_remote_worker.cc | 307 TF_CHECK_OK(ReadInt64FromEnvVar("GRPC_MAX_RETRIES", 0, &max_retries)); in MaxRetries()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | env_var.h | 36 Status ReadInt64FromEnvVar(StringPiece env_var_name, int64_t default_val,
|
| D | use_cudnn.cc | 104 Status status = ReadInt64FromEnvVar(#flag_name, default_value, &value); \
|
| D | env_var.cc | 48 Status ReadInt64FromEnvVar(StringPiece env_var_name, int64_t default_val, in ReadInt64FromEnvVar() function
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | process_state.cc | 95 Status status = ReadInt64FromEnvVar("TF_CPU_BFC_MEM_LIMIT_IN_MB", in GetCPUAllocator()
|
| /external/tensorflow/tensorflow/core/common_runtime/pluggable_device/ |
| D | pluggable_device_process_state.cc | 190 Status status = ReadInt64FromEnvVar("TF_GPU_HOST_MEM_LIMIT_IN_MB", in GetPluggableDeviceHostAllocator()
|
| D | pluggable_device.cc | 216 ReadInt64FromEnvVar("TF_GPU_THREAD_COUNT", 2, &device_thread_count)); in Init()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | algorithm_selector.cc | 179 Status status = tensorflow::ReadInt64FromEnvVar("TF_TRT_FIXED_ALGORITHM_ID", in GetFixedAlgorithmID()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | matmul_util.cc | 72 ReadInt64FromEnvVar("TF_MATMUL_AUTOTUNE_MAX_ALGORITHMS", 10, &value); in MatmulMaxAutotuneAlgorithmCount()
|
| D | fft_ops.cc | 418 Status status = ReadInt64FromEnvVar(envvar_in_mb, default_value_in_bytes, in GetCufftWorkspaceLimit()
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_cudamallocasync_allocator.cc | 406 TF_CHECK_OK(ReadInt64FromEnvVar("TF_CUDA_MALLOC_ASYNC_SUPPORTED_PREALLOC", 0, in SetStreamAndPreallocateMemory()
|
| D | gpu_process_state.cc | 349 Status status = ReadInt64FromEnvVar("TF_GPU_HOST_MEM_LIMIT_IN_MB", in GetGpuHostAllocator()
|
| D | gpu_device.cc | 519 ReadInt64FromEnvVar("TF_GPU_THREAD_COUNT", 2, &gpu_thread_count)); in Init()
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
| D | stream_executor_pimpl.cc | 134 SE_CHECK_OK(tensorflow::ReadInt64FromEnvVar("TF_PER_DEVICE_MEMORY_LIMIT_MB", in GetMemoryLimitBytes()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/segment/ |
| D | segment.cc | 1219 TF_CHECK_OK(ReadInt64FromEnvVar("TF_TRT_MAX_ALLOWED_ENGINES", in SegmentGraph()
|
| /external/tensorflow/tensorflow/core/util/tensor_bundle/ |
| D | tensor_bundle.cc | 779 ReadInt64FromEnvVar("TF_TABLE_INDEX_CACHE_SIZE_IN_MB", 0, &cache_size); in BundleReader()
|