| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| gpu_bfc_allocator.cc | 30 const GPUOptions& gpu_options,
35 total_memory, gpu_options.allow_growth(), name) {}
|
| gpu_event_mgr.cc | 25 EventMgr::EventMgr(gpu::StreamExecutor* se, const GPUOptions& gpu_options)
27 deferred_bytes_threshold_(gpu_options.deferred_deletion_bytes()
28 ? gpu_options.deferred_deletion_bytes()
30 polling_active_delay_usecs_(gpu_options.polling_active_delay_usecs()
31 ? gpu_options.polling_active_delay_usecs()
34 gpu_options.polling_inactive_delay_msecs()
35 ? gpu_options.polling_inactive_delay_msecs()
|
| gpu_bfc_allocator.h | 45 const GPUOptions& gpu_options, const string& name);
|
| gpu_device_factory.cc | 38 options.config.gpu_options().force_gpu_compatible();
87 options.config.gpu_options().force_gpu_compatible();
|
| gpu_device_test.cc | 37 GPUOptions* gpu_options = config->mutable_gpu_options(); local
38 gpu_options->set_visible_device_list(visible_device_list);
39 gpu_options->set_per_process_gpu_memory_fraction(
43 gpu_options->mutable_experimental()->add_virtual_devices();
|
| gpu_device.cc | 279 em_.reset(new EventMgr(executor_, options.config.gpu_options()));
710 const size_t num_gpus_to_use, const GPUOptions& gpu_options,
713 const auto& virtual_devices = gpu_options.experimental().virtual_devices();
715 if (gpu_options.per_process_gpu_memory_fraction() > 0) {
726 if (!gpu_options.visible_device_list().empty() &&
782 Status SingleVirtualDeviceMemoryLimit(const GPUOptions& gpu_options,
796 gpu_options.per_process_gpu_memory_fraction();
864 const auto& gpu_options = options.config.gpu_options(); local
[all...] |
| gpu_event_mgr.h | 50 const GPUOptions& gpu_options);
|
| /external/tensorflow/tensorflow/contrib/mpi_collectives/ |
| mpi_allgather_test.py | 100 gpu_options = tf.GPUOptions(visible_device_list=str(my_rank))
101 config = tf.ConfigProto(gpu_options=gpu_options)
|
| mpi_allreduce_test.py | 105 gpu_options = tf.GPUOptions(visible_device_list=str(my_rank))
106 config = tf.ConfigProto(gpu_options=gpu_options)
|
| 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/tensorrt/test/ |
| test_tftrt.py | 63 gpu_options = cpb2.GPUOptions(per_process_gpu_memory_fraction=0.50)
72 config=cpb2.ConfigProto(gpu_options=gpu_options), graph=g) as sess:
|
| /external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/ |
| run_config.py | 283 gpu_options = config_pb2.GPUOptions(
289 gpu_options=gpu_options)
|
| /external/tensorflow/tensorflow/tools/dist_test/server/ |
| grpc_tensorflow_server.py | 107 config = config_pb2.ConfigProto(gpu_options=config_pb2.GPUOptions(
|
| /external/tensorflow/tensorflow/python/training/ |
| server_lib_test.py | 182 gpu_options=config_pb2.GPUOptions(per_process_gpu_memory_fraction=0.1))
186 self.assertEqual(0.1, server.server_def.default_session_config.gpu_options.
200 self.assertEqual(0.1, server.server_def.default_session_config.gpu_options.
|
| /external/tensorflow/tensorflow/python/debug/lib/ |
| dist_session_debug_grpc_test.py | 87 gpu_options=config_pb2.GPUOptions(
|
| grpc_debug_test_server.py | 451 config = config_pb2.ConfigProto(gpu_options=config_pb2.GPUOptions(
|
| /external/tensorflow/tensorflow/python/client/ |
| virtual_gpu_test.py | 96 return config_pb2.ConfigProto(gpu_options=virtual_device_gpu_options)
|
| session.py | [all...] |
| /external/tensorflow/tensorflow/python/estimator/ |
| training.py | 680 gpu_options=config.session_config.gpu_options)
[all...] |
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| sparse_xent_op_test.py | 291 config.gpu_options.per_process_gpu_memory_fraction = 0.3
|
| fft_ops_test.py | 67 config.gpu_options.per_process_gpu_memory_fraction = 1e-2
|
| /external/tensorflow/tensorflow/contrib/learn/python/learn/ |
| experiment_test.py | 314 expected_config_proto.gpu_options.per_process_gpu_memory_fraction = 0.314
[all...] |
| /external/tensorflow/tensorflow/python/framework/ |
| test_util.py | 858 config.gpu_options.per_process_gpu_memory_fraction = 0.3
[all...] |
