| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | master.cc | 138 const protobuf::RepeatedPtrField<string>& device_filters, MasterEnv* env, in GetRemoteDevices() argument
141 DeviceFinder finder(device_filters, env, worker_cache); in GetRemoteDevices()
149 const protobuf::RepeatedPtrField<string>& device_filters, MasterEnv* env, in GetRemoteWorkers() argument
151 DeviceFinder finder(device_filters, env, worker_cache); in GetRemoteWorkers()
157 const protobuf::RepeatedPtrField<string>& device_filters, MasterEnv* env, in DeviceFinder() argument
169 for (const string& filter : device_filters) { in DeviceFinder()
214 if (device_filters.empty()) { in DeviceFinder()
217 for (const string& filter : device_filters) { in DeviceFinder()
423 DeviceFinder::GetRemoteDevices(req->config().device_filters(), env_, in CreateSession()
441 DeviceFinder::GetRemoteDevices(req->config().device_filters(), env_, in CreateSession()
[all …]
|
| /external/tensorflow/tensorflow/python/training/ |
| D | server_lib.py | 533 def set_device_filters(self, job_name, task_index, device_filters): argument
535 assert all(isinstance(df, str) for df in device_filters)
537 self._device_filters[job_name][task_index] = [df for df in device_filters]
574 jdf.tasks[i].device_filters.append(tdf)
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.config.experimental.-cluster-device-filters.pbtxt | 11 …argspec: "args=[\'self\', \'job_name\', \'task_index\', \'device_filters\'], varargs=None, keyword…
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.config.experimental.-cluster-device-filters.pbtxt | 11 …argspec: "args=[\'self\', \'job_name\', \'task_index\', \'device_filters\'], varargs=None, keyword…
|
| D | tensorflow.-config-proto.pbtxt | 44 name: "device_filters"
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | parameter_server_strategy_v2.py | 496 device_filters = server_lib.ClusterDeviceFilters()
499 device_filters.set_device_filters(
504 device_filters.set_device_filters(
522 cluster_device_filters=device_filters)
|
| D | parameter_server_strategy.py | 627 del updated_config.device_filters[:]
629 updated_config.device_filters.extend(
632 updated_config.device_filters.append(
|
| D | distribute_coordinator_test.py | 123 session_config.device_filters.extend(
127 session_config.device_filters.append("/job:somejob")
821 self._device_filters.extend(session_config.device_filters)
831 self._device_filters.extend(session_config.device_filters)
|
| D | collective_all_reduce_strategy.py | 483 device_filters=("/job:%s/task:%d" % (task_type, task_id),))
778 del updated_config.device_filters[:]
779 updated_config.device_filters.append(
|
| D | collective_all_reduce_strategy_test.py | 404 config_proto = config_pb2.ConfigProto(device_filters=['to_be_overridden'])
415 self.assertEqual(['/job:worker/task:1'], new_config.device_filters)
|
| D | parameter_server_strategy_test.py | 708 config_proto = config_pb2.ConfigProto(device_filters=['to_be_overridden'])
714 new_config.device_filters)
|
| D | distribute_coordinator.py | 548 del session_config.device_filters[:]
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | context_distributed_manager.cc | 654 std::vector<string> device_filters(tdf.second.device_filters_size()); in SetOrUpdateServerDef() local
656 device_filters[i] = tdf.second.device_filters(i); in SetOrUpdateServerDef()
661 context_->SetRemoteDeviceFilters(remote_worker, device_filters)); in SetOrUpdateServerDef()
|
| D | context.cc | 1372 const string& remote_worker, const std::vector<string>& device_filters) { in SetRemoteDeviceFilters() argument
1390 for (auto& filter : device_filters) { in SetRemoteDeviceFilters()
1401 for (auto& filter : device_filters) { in SetRemoteDeviceFilters()
|
| D | context.h | 465 const std::vector<string>& device_filters);
|
| /external/tensorflow/tensorflow/core/protobuf/ |
| D | tensorflow_server.proto | 22 import "tensorflow/core/protobuf/device_filters.proto";
|
| D | device_filters.proto | 58 repeated string device_filters = 1; field
|
| D | BUILD | 34 "device_filters.proto",
|
| D | master.proto | 248 repeated string device_filters = 2; field
|
| D | config.proto | 505 repeated string device_filters = 4; field
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | context.py | 833 device_filters=None): argument
858 self._collective_device_filters != device_filters):
869 self._collective_device_filters = device_filters
1169 del config.device_filters[:]
1171 config.device_filters.append(f)
|
| /external/tensorflow/tensorflow/python/keras/distribute/ |
| D | distribute_coordinator_utils.py | 438 del session_config.device_filters[:]
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | config_test.py | 842 device_filters=['/job:worker/task:1'])
850 self.assertEqual(['/job:worker/task:1'], new_config.device_filters)
|
| /external/tensorflow/ |
| D | Android.bp | 258 "tensorflow/core/protobuf/device_filters.proto",
|
| /external/tensorflow/tensorflow/core/ |
| D | BUILD | 186 "//tensorflow/core/protobuf:device_filters.proto",
|