| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_bfc_allocator.cc | 22 bool GPUBFCAllocator::GetAllowGrowthValue(const GPUOptions& gpu_options) { in GetAllowGrowthValue() argument
26 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()
82 const GPUOptions& gpu_options, in GPUBFCAllocator() argument
85 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 polling_active_delay_usecs_(gpu_options.polling_active_delay_usecs() in EventMgr()
95 ? gpu_options.polling_active_delay_usecs() in EventMgr()
251 const GPUOptions& gpu_options) { in GetEventMgr() argument
259 auto event_mgr = new EventMgr(se, gpu_options); in GetEventMgr()
|
| D | gpu_bfc_allocator.h | 40 const GPUOptions& gpu_options, const string& name);
46 static bool GetAllowGrowthValue(const GPUOptions& gpu_options);
|
| D | gpu_device.cc | 435 tf_gpu_id_, 0, executor_, options.config.gpu_options()); in Init()
445 options.config.gpu_options()); in Init()
448 options.config.gpu_options().experimental().kernel_tracker_max_interval(), in Init()
449 options.config.gpu_options().experimental().kernel_tracker_max_bytes(), in Init()
450 options.config.gpu_options().experimental().kernel_tracker_max_pending()); in Init()
452 options.config.gpu_options().experimental().timestamped_allocator(); in Init()
827 const size_t num_gpus_to_use, const GPUOptions& gpu_options, in VerifyVirtualDeviceSettings() argument
831 const auto& virtual_devices = gpu_options.experimental().virtual_devices(); in VerifyVirtualDeviceSettings()
833 if (gpu_options.per_process_gpu_memory_fraction() > 0) { in VerifyVirtualDeviceSettings()
844 if (!gpu_options.visible_device_list().empty() && in VerifyVirtualDeviceSettings()
[all …]
|
| D | gpu_device_factory.cc | 40 options.config.gpu_options().force_gpu_compatible(); in GPUDevice()
90 options.config.gpu_options().force_gpu_compatible(); in GPUCompatibleCPUDevice()
|
| D | gpu_device_test.cc | 78 GPUOptions* gpu_options = config->mutable_gpu_options(); local
79 gpu_options->set_visible_device_list(visible_device_list);
80 gpu_options->set_per_process_gpu_memory_fraction(
84 gpu_options->mutable_experimental()->add_virtual_devices();
|
| /external/tensorflow/tensorflow/core/common_runtime/device/ |
| D | device_event_mgr.cc | 93 EventMgr::EventMgr(se::StreamExecutor* se, const GPUOptions& gpu_options) in EventMgr() argument
95 polling_active_delay_usecs_(gpu_options.polling_active_delay_usecs() in EventMgr()
96 ? gpu_options.polling_active_delay_usecs() in EventMgr()
252 const GPUOptions& gpu_options) { in GetEventMgr() argument
260 auto event_mgr = new EventMgr(se, gpu_options); in GetEventMgr()
|
| D | device_event_mgr.h | 99 EventMgr(se::StreamExecutor* se, const GPUOptions& gpu_options);
153 EventMgr* GetEventMgr(se::StreamExecutor* se, const GPUOptions& gpu_options);
|
| D | device_event_mgr_test.cc | 42 TEST_EventMgr(se::StreamExecutor* se, const GPUOptions& gpu_options) in TEST_EventMgr() argument
43 : EventMgr(se, gpu_options) {} in TEST_EventMgr()
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | config_test.py | 582 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)
738 gpu_options=config_pb2.GPUOptions(visible_device_list='0'))
740 self.assertEqual(new_config.gpu_options.visible_device_list, '0')
759 gpu_options=config_pb2.GPUOptions(visible_device_list=''))
763 self.assertEqual(new_config.gpu_options.visible_device_list,
769 gpu_options=config_pb2.GPUOptions(visible_device_list=str(gpu_count)))
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_executable.cc | 184 const GpuExecutableRunOptions* gpu_options = in ExecuteThunks() local
193 gpu_options && gpu_options->gpu_global_device_ids() in ExecuteThunks()
194 ? &*gpu_options->gpu_global_device_ids() in ExecuteThunks()
196 gpu_options && gpu_options->nccl_unique_id_callback() in ExecuteThunks()
197 ? &gpu_options->nccl_unique_id_callback() in ExecuteThunks()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_graph_test.cc | 90 GPUOptions* gpu_options = config->mutable_gpu_options(); in TEST() local
92 gpu_options->mutable_experimental()->add_virtual_devices(); in TEST()
|
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| D | xrt_execute_op.cc | 325 xla::gpu::GpuExecutableRunOptions gpu_options; in RunExecutable() local
332 gpu_options.set_gpu_global_device_ids(gpu_global_ids); in RunExecutable()
344 gpu_options.set_nccl_unique_id_callback(uid_callback); in RunExecutable()
346 run_options.set_gpu_executable_run_options(&gpu_options); in RunExecutable()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | collective_executor_mgr.cc | 35 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 | 89 gpu_options=config_pb2.GPUOptions(
|
| /external/tensorflow/tensorflow/python/training/ |
| D | server_lib_test.py | 185 gpu_options=config_pb2.GPUOptions(per_process_gpu_memory_fraction=0.1))
189 self.assertEqual(0.1, server.server_def.default_session_config.gpu_options.
203 self.assertEqual(0.1, server.server_def.default_session_config.gpu_options.
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_gpu_device.cc | 138 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/compiler/tensorrt/test/ |
| D | tf_trt_integration_test_base.py | 328 gpu_options = config_pb2.GPUOptions()
329 gpu_options.allow_growth = True
330 return gpu_options
349 gpu_options=self._GetGPUOptions(),
|
| D | quantization_mnist_test.py | 239 config_proto.gpu_options.allow_growth = True
|
| /external/libopus/training/ |
| D | rnn_train.py | 25 config.gpu_options.per_process_gpu_memory_fraction = 0.44
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | context.py | 1025 gpu_options = self._compute_gpu_options()
1026 config.gpu_options.MergeFrom(gpu_options)
1378 visible_indices = self._config.gpu_options.visible_device_list
|
| /external/tensorflow/tensorflow/python/client/ |
| D | virtual_gpu_test.py | 96 return config_pb2.ConfigProto(gpu_options=virtual_device_gpu_options)
|
| D | session.py | 1753 gpu_options = config_pb2.GPUOptions(allow_growth=True)
1754 config = config_pb2.ConfigProto(gpu_options=gpu_options)
|