Searched refs:device_filters (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | master.cc | 136 const protobuf::RepeatedPtrField<string>& device_filters, MasterEnv* env, in GetRemoteDevices() argument
139 DeviceFinder finder(device_filters, env, worker_cache); in GetRemoteDevices()
147 const protobuf::RepeatedPtrField<string>& device_filters, MasterEnv* env, in GetRemoteWorkers() argument
149 DeviceFinder finder(device_filters, env, worker_cache); in GetRemoteWorkers()
155 const protobuf::RepeatedPtrField<string>& device_filters, MasterEnv* env, in DeviceFinder() argument
167 for (const string& filter : device_filters) { in DeviceFinder()
212 if (device_filters.empty()) { in DeviceFinder()
215 for (const string& filter : device_filters) { in DeviceFinder()
426 DeviceFinder::GetRemoteDevices(req->config().device_filters(), env_, in CreateSession()
444 DeviceFinder::GetRemoteDevices(req->config().device_filters(), env_, in CreateSession()
[all …]
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.-config-proto.pbtxt | 44 name: "device_filters"
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | collective_all_reduce_strategy.py | 424 del updated_config.device_filters[:]
425 updated_config.device_filters.append(
|
| D | distribute_coordinator_test.py | 131 session_config.device_filters.extend(
135 session_config.device_filters.append("/job:somejob")
827 self._device_filters.extend(session_config.device_filters)
837 self._device_filters.extend(session_config.device_filters)
|
| D | parameter_server_strategy.py | 507 del updated_config.device_filters[:]
508 updated_config.device_filters.extend(
|
| D | distribute_coordinator.py | 549 del session_config.device_filters[:]
|
| /external/tensorflow/tensorflow/contrib/distribute/python/ |
| D | collective_all_reduce_strategy_test.py | 464 config_proto = config_pb2.ConfigProto(device_filters=['to_be_overridden'])
475 self.assertEqual(['/job:worker/task:1'], new_config.device_filters)
|
| D | parameter_server_strategy_test.py | 786 config_proto = config_pb2.ConfigProto(device_filters=['to_be_overridden'])
792 new_config.device_filters)
|
| /external/tensorflow/tensorflow/core/protobuf/ |
| D | master.proto | 248 repeated string device_filters = 2; field
|
| D | config.proto | 377 repeated string device_filters = 4; field
|
| /external/tensorflow/ |
| D | RELEASE.md | 312 * `RunConfig` now sets device_filters to restrict how workers and PS can communicate. This can spee…
|