| /external/tensorflow/tensorflow/contrib/session_bundle/ |
| session_bundle_test.py | 94 config=config_pb2.ConfigProto(device_count={"CPU": 2}))
122 config=config_pb2.ConfigProto(device_count={"CPU": 2}))
132 config=config_pb2.ConfigProto(device_count={"CPU": 2}))
|
| exporter_test.py | 63 config=config_pb2.ConfigProto(device_count={"CPU": 2})) as sess:
137 config=config_pb2.ConfigProto(device_count={"CPU": 2})) as sess:
|
| bundle_shim_test.py | 324 config=config_pb2.ConfigProto(device_count={"CPU": 2})))
348 config=config_pb2.ConfigProto(device_count={"CPU": 2})))
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| hierarchical_tree_broadcaster.cc | 130 int device_count = 0; local
142 perm.push_back(device_count);
144 col_params->instance.device_names[device_count] == device_name;
147 device_count += dev_per_task[ti];
|
| function_threadpool_test.cc | 55 auto* device_count = options.config.mutable_device_count(); local
56 device_count->insert({"CPU", 3});
|
| partitioning_utils_test.cc | 43 auto* device_count = options.config.mutable_device_count(); variable
44 device_count->insert({"CPU", 2});
|
| collective_param_resolver_local_test.cc | 38 auto* device_count = options.config.mutable_device_count(); local
39 device_count->insert({"CPU", NUM_DEVS});
|
| /external/tensorflow/tensorflow/python/data/kernel_tests/ |
| memory_cleanup_test.py | 107 config=config_pb2.ConfigProto(device_count={"CPU": 3, "GPU": 1}))
|
| iterator_cluster_test.py | 45 worker_config.device_count["CPU"] = 2
101 worker_config.device_count["CPU"] = 2
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| save_restore_ops_test.py | 48 target="", config=config_pb2.ConfigProto(device_count={"CPU": 2})):
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| rpc_collective_executor_mgr_test.cc | 43 auto* device_count = options.config.mutable_device_count(); local
44 device_count->insert({"CPU", NUM_DEVS});
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| service.cc | 164 if (execute_backend_->device_count() > 0) {
165 CHECK_GE(execute_backend_->device_count(), options_.number_of_replicas())
172 for (int i = 0; i < execute_backend_->device_count(); ++i) {
631 execute_backend_->device_count()) {
761 const int64 available_device_count = execute_backend_->device_count();
766 if (available_device_count < arg->device_count() * replica_count) {
770 arg->device_count(), replica_count, available_device_count);
773 for (int64 i = 0; i < arg->device_count(); ++i) {
[all...] |
| /external/tensorflow/tensorflow/python/debug/cli/ |
| profile_analyzer_cli.py | 417 device_count = len(self._run_metadata.step_stats.dev_stats)
418 for index in range(device_count):
433 device_stats.device, index, device_count,
477 self, device_name, device_index, device_count,
486 device_count: (int) Number of devices.
536 device_index + 1, device_count, device_name)
622 device_count = len(self._run_metadata.step_stats.dev_stats)
623 for index in range(device_count):
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| client.cc | 292 GetDeviceHandles(/*device_count=*/1));
366 int64 device_count) {
367 if (device_count < 1) {
368 return InvalidArgument("device_count must be greater than 0");
371 request.set_device_count(device_count);
|
| client.h | 122 // Requests device_count device handles available on the target. The returned
126 StatusOr<std::vector<DeviceHandle>> GetDeviceHandles(int64 device_count);
|
| local_client.cc | 224 int LocalClient::device_count() const { function in class:xla::LocalClient
225 return local_service_->backend().device_count();
|
| /external/tensorflow/tensorflow/python/eager/ |
| function_defun_collection_test.py | 101 config=config_pb2.ConfigProto(device_count={'CPU': 4}))
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| xla.proto | 309 int64 device_count = 1;
|
| xla_data.proto | 313 int64 device_count = 2;
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| dataset_test_base.cc | 145 auto* device_count = options.config.mutable_device_count(); local
146 device_count->insert({"CPU", cpu_num});
|
| /external/tensorflow/tensorflow/python/client/ |
| timeline_test.py | 143 config = config_pb2.ConfigProto(device_count={'CPU': 3})
173 config = config_pb2.ConfigProto(device_count={'CPU': 3})
|
| /external/tensorflow/tensorflow/python/profiler/ |
| pprof_profiler.py | 293 device_count = len(self._run_metadata.step_stats.dev_stats)
296 (device_index + 1, device_count, device_stats.device))
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| rocm_driver.cc | 1042 int device_count = 0; local
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| local_client_execute_test.cc | 607 for (int d = 0; d < local_client_->device_count(); ++d) {
638 local_client_->device_count()),
640 local_client_->device_count()));
652 for (int d = 0; d < local_client_->device_count(); ++d) {
[all...] |
| /external/tensorflow/tensorflow/lite/experimental/examples/lstm/ |
| bidirectional_sequence_lstm_test.py | 33 CONFIG = tf.ConfigProto(device_count={"GPU": 0})
|
