Home
last modified time | relevance | path

Searched refs:allow_growth (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_bfc_allocator.cc26 return gpu_options.allow_growth(); in GetAllowGrowthValue()
30 if (gpu_options.allow_growth()) { in GetAllowGrowthValue()
34 << " config value was " << gpu_options.allow_growth() << "."; in GetAllowGrowthValue()
38 if (!gpu_options.allow_growth()) { in GetAllowGrowthValue()
42 << " config value was " << gpu_options.allow_growth() << "."; in GetAllowGrowthValue()
51 << " of " << gpu_options.allow_growth() << "."; in GetAllowGrowthValue()
52 return gpu_options.allow_growth(); in GetAllowGrowthValue()
/external/tensorflow/tensorflow/python/framework/
Dconfig_test.py582 self.assertTrue(c.gpu_options.allow_growth)
663 self.assertTrue(c.gpu_options.allow_growth)
668 self.assertFalse(c.gpu_options.allow_growth)
679 self.assertTrue(c.gpu_options.allow_growth)
694 self.assertFalse(c.gpu_options.allow_growth)
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.-g-p-u-options.pbtxt12 name: "allow_growth"
/external/tensorflow/tensorflow/python/eager/
Dcontext.py1085 allow_growth = memory_growths.pop()
1087 allow_growth = None
1090 allow_growth=allow_growth,
/external/tensorflow/tensorflow/python/compiler/tensorrt/test/
Dquantization_mnist_test.py239 config_proto.gpu_options.allow_growth = True
Dtf_trt_integration_test_base.py329 gpu_options.allow_growth = True
/external/tensorflow/tensorflow/core/common_runtime/
Dbfc_allocator.h53 bool allow_growth, const string& name,
Dbfc_allocator.cc41 bool allow_growth, const string& name, in BFCAllocator() argument
49 if (allow_growth) { in BFCAllocator()
/external/tensorflow/tensorflow/python/compiler/tensorrt/
Dtrt_convert_test.py84 gpu_options=config_pb2.GPUOptions(allow_growth=True))
/external/tensorflow/tensorflow/python/client/
Dsession.py1753 gpu_options = config_pb2.GPUOptions(allow_growth=True)
/external/tensorflow/tensorflow/core/protobuf/
Dconfig.proto23 // GPU memory is pre-allocated unless the allow_growth option is enabled.
40 bool allow_growth = 4; field