| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_bfc_allocator.cc | 25 bool GPUBFCAllocator::GetAllowGrowthValue(const GPUOptions& gpu_options) { in GetAllowGrowthValue() argument
29 return gpu_options.allow_growth(); in GetAllowGrowthValue()
33 if (gpu_options.allow_growth()) { in GetAllowGrowthValue()
37 << " config value was " << gpu_options.allow_growth() << "."; in GetAllowGrowthValue()
41 if (!gpu_options.allow_growth()) { in GetAllowGrowthValue()
45 << " config value was " << gpu_options.allow_growth() << "."; in GetAllowGrowthValue()
54 << " of " << gpu_options.allow_growth() << "."; in GetAllowGrowthValue()
55 return gpu_options.allow_growth(); in GetAllowGrowthValue()
64 const GPUOptions& gpu_options, in GPUBFCAllocator() argument
67 GPUBFCAllocator::GetAllowGrowthValue(gpu_options), name) {} in GPUBFCAllocator()
|
| D | gpu_event_mgr.cc | 92 EventMgr::EventMgr(se::StreamExecutor* se, const GPUOptions& gpu_options) in EventMgr() argument
94 deferred_bytes_threshold_(gpu_options.deferred_deletion_bytes() in EventMgr()
95 ? gpu_options.deferred_deletion_bytes() in EventMgr()
97 polling_active_delay_usecs_(gpu_options.polling_active_delay_usecs() in EventMgr()
98 ? gpu_options.polling_active_delay_usecs() in EventMgr()
|
| D | gpu_device.cc | 314 pending_cap_ = options.config.gpu_options().experimental().pending_cap(); in BaseGPUDevice()
316 options.config.gpu_options().experimental().timestamped_allocator(); in BaseGPUDevice()
380 em_.reset(new EventMgr(executor_, options.config.gpu_options())); in Init()
389 tf_gpu_id_, i, executor_, options.config.gpu_options())); in Init()
815 const size_t num_gpus_to_use, const GPUOptions& gpu_options, in VerifyVirtualDeviceSettings() argument
818 const auto& virtual_devices = gpu_options.experimental().virtual_devices(); in VerifyVirtualDeviceSettings()
820 if (gpu_options.per_process_gpu_memory_fraction() > 0) { in VerifyVirtualDeviceSettings()
831 if (!gpu_options.visible_device_list().empty() && in VerifyVirtualDeviceSettings()
887 Status SingleVirtualDeviceMemoryLimit(const GPUOptions& gpu_options, in SingleVirtualDeviceMemoryLimit() argument
901 gpu_options.per_process_gpu_memory_fraction(); in SingleVirtualDeviceMemoryLimit()
[all …]
|
| D | gpu_bfc_allocator.h | 92 const GPUOptions& gpu_options, const string& name);
98 static bool GetAllowGrowthValue(const GPUOptions& gpu_options);
|
| D | gpu_device_test.cc | 75 GPUOptions* gpu_options = config->mutable_gpu_options(); local
76 gpu_options->set_visible_device_list(visible_device_list);
77 gpu_options->set_per_process_gpu_memory_fraction(
81 gpu_options->mutable_experimental()->add_virtual_devices();
|
| D | gpu_device_factory.cc | 39 options.config.gpu_options().force_gpu_compatible(); in GPUDevice()
89 options.config.gpu_options().force_gpu_compatible(); in GPUCompatibleCPUDevice()
|
| D | gpu_event_mgr.h | 66 EventMgr(se::StreamExecutor* se, const GPUOptions& gpu_options);
|
| /external/tensorflow/tensorflow/contrib/tensorrt/custom_plugin_examples/ |
| D | plugin_test.py | 60 gpu_options = config_pb2.GPUOptions(per_process_gpu_memory_fraction=0.50)
70 config=config_pb2.ConfigProto(gpu_options=gpu_options),
|
| /external/tensorflow/tensorflow/contrib/mpi_collectives/ |
| D | mpi_allgather_test.py | 100 gpu_options = tf.GPUOptions(visible_device_list=str(my_rank))
101 config = tf.ConfigProto(gpu_options=gpu_options)
|
| D | mpi_allreduce_test.py | 105 gpu_options = tf.GPUOptions(visible_device_list=str(my_rank))
106 config = tf.ConfigProto(gpu_options=gpu_options)
|
| D | mpi_ops_test.py | 124 cpu_config = tf.ConfigProto(gpu_options=no_gpus)
129 gpu_config = tf.ConfigProto(gpu_options=one_gpu)
|
| /external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/ |
| D | run_config.py | 308 gpu_options = config_pb2.GPUOptions(
314 gpu_options=gpu_options)
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_graph_test.cc | 163 GPUOptions* gpu_options = config->mutable_gpu_options(); in TEST() local
165 gpu_options->mutable_experimental()->add_virtual_devices(); in TEST()
|
| /external/tensorflow/tensorflow/python/compiler/tensorrt/test/ |
| D | tf_trt_integration_test_base.py | 213 gpu_options = config_pb2.GPUOptions()
214 gpu_options.allow_growth = True
215 return gpu_options
241 gpu_options=self._GetGPUOptions(), graph_options=graph_options)
|
| D | quantization_mnist_test.py | 244 config_proto.gpu_options.allow_growth = True
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | context.py | 735 return self._config.gpu_options.per_process_gpu_memory_fraction
743 self._config.gpu_options.per_process_gpu_memory_fraction = fraction
747 return self._config.gpu_options.allow_growth
755 self._config.gpu_options.allow_growth = enabled
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | collective_executor_mgr.cc | 34 config.gpu_options().experimental().collective_ring_order()) {} in CollectiveExecutorMgr()
|
| /external/tensorflow/tensorflow/python/debug/lib/ |
| D | grpc_tensorflow_server.py | 108 config = config_pb2.ConfigProto(gpu_options=config_pb2.GPUOptions(
|
| D | dist_session_debug_grpc_test.py | 87 gpu_options=config_pb2.GPUOptions(
|
| /external/tensorflow/tensorflow/python/training/ |
| D | server_lib_test.py | 186 gpu_options=config_pb2.GPUOptions(per_process_gpu_memory_fraction=0.1))
190 self.assertEqual(0.1, server.server_def.default_session_config.gpu_options.
204 self.assertEqual(0.1, server.server_def.default_session_config.gpu_options.
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_gpu_device.cc | 83 session_options.config.gpu_options().visible_device_list(); in CreateDevices()
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.-config-proto.pbtxt | 50 name: "gpu_options"
|
| /external/tensorflow/tensorflow/python/client/ |
| D | session.py | 1709 gpu_options = config_pb2.GPUOptions(allow_growth=True)
1710 config = config_pb2.ConfigProto(gpu_options=gpu_options)
|
| D | virtual_gpu_test.py | 96 return config_pb2.ConfigProto(gpu_options=virtual_device_gpu_options)
|
| /external/tensorflow/tensorflow/contrib/distribute/python/ |
| D | multi_worker_test_base.py | 141 worker_config.gpu_options.per_process_gpu_memory_fraction = gpu_mem_frac
|