diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2018-09-19 17:28:57 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-19 17:32:07 -0700 |
| commit | 5d2047029a77545c97c0fdf74d9c03c92d1dcb88 (patch) | |
| tree | b5a232646fe6c0b976104c6462709b71cb01eee7 /tensorflow/core/common_runtime/gpu/gpu_bfc_allocator.h | |
| parent | dfa0447740b42edff5ae2d76d5957aa688ae8053 (diff) | |
Allow setting a global override for the "allow_growth" GPU option via the TF_FORCE_GPU_ALLOW_GROWTH environment variable.
PiperOrigin-RevId: 213728460
Diffstat (limited to 'tensorflow/core/common_runtime/gpu/gpu_bfc_allocator.h')
| -rw-r--r-- | tensorflow/core/common_runtime/gpu/gpu_bfc_allocator.h | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/tensorflow/core/common_runtime/gpu/gpu_bfc_allocator.h b/tensorflow/core/common_runtime/gpu/gpu_bfc_allocator.h index 3470f7a9f7..d4c9cee89a 100644 --- a/tensorflow/core/common_runtime/gpu/gpu_bfc_allocator.h +++ b/tensorflow/core/common_runtime/gpu/gpu_bfc_allocator.h @@ -93,6 +93,9 @@ class GPUBFCAllocator : public BFCAllocator { ~GPUBFCAllocator() override {} TF_DISALLOW_COPY_AND_ASSIGN(GPUBFCAllocator); + + private: + static bool GetAllowGrowthValue(const GPUOptions& gpu_options); }; } // namespace tensorflow |