Searched refs:ReadInt64FromEnvVar (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/core/util/ |
| D | env_var.h | 36 Status ReadInt64FromEnvVar(StringPiece env_var_name, int64 default_val,
|
| D | use_cudnn.cc | 62 Status status = ReadInt64FromEnvVar(#flag_name, default_value, &value); \
|
| D | env_var.cc | 48 Status ReadInt64FromEnvVar(StringPiece env_var_name, int64 default_val, in ReadInt64FromEnvVar() function
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | process_state.cc | 87 Status status = ReadInt64FromEnvVar("TF_CPU_BFC_MEM_LIMIT_IN_MB", in GetCPUAllocator()
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_process_state.cc | 236 Status status = ReadInt64FromEnvVar("TF_GPU_HOST_MEM_LIMIT_IN_MB", in GetGpuHostAllocator()
|
| D | gpu_device.cc | 422 ReadInt64FromEnvVar("TF_GPU_THREAD_COUNT", 2, &gpu_thread_count)); in Init()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fft_ops.cc | 358 Status status = ReadInt64FromEnvVar(envvar_in_mb, default_value_in_bytes, in GetCufftWorkspaceLimit()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream_executor_pimpl.cc | 175 SE_CHECK_OK(tensorflow::ReadInt64FromEnvVar("TF_PER_DEVICE_MEMORY_LIMIT_MB", in GetMemoryLimitBytes()
|