| /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/python/training/ |
| D | server_lib.py | 532 def set_device_filters(self, job_name, task_index, device_filters): argument
534 assert all(isinstance(df, str) for df in device_filters)
536 self._device_filters[job_name][task_index] = [df for df in device_filters]
573 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 | collective_all_reduce_strategy.py | 274 device_filters=("/job:%s/task:%d" % (task_type, task_id),))
502 del updated_config.device_filters[:]
503 updated_config.device_filters.append(
|
| D | parameter_server_strategy.py | 594 del updated_config.device_filters[:]
596 updated_config.device_filters.extend(
599 updated_config.device_filters.append(
|
| D | distribute_coordinator_test.py | 134 session_config.device_filters.extend(
138 session_config.device_filters.append("/job:somejob")
840 self._device_filters.extend(session_config.device_filters)
850 self._device_filters.extend(session_config.device_filters)
|
| D | collective_all_reduce_strategy_test.py | 481 config_proto = config_pb2.ConfigProto(device_filters=['to_be_overridden'])
492 self.assertEqual(['/job:worker/task:1'], new_config.device_filters)
|
| D | parameter_server_strategy_test.py | 705 config_proto = config_pb2.ConfigProto(device_filters=['to_be_overridden'])
711 new_config.device_filters)
|
| D | distribute_coordinator.py | 552 del session_config.device_filters[:]
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | context.py | 650 device_filters=None): argument
675 self._collective_device_filters != device_filters):
686 self._collective_device_filters = device_filters
939 del config.device_filters[:]
941 config.device_filters.append(f)
|
| /external/tensorflow/tensorflow/core/protobuf/ |
| D | tensorflow_server.proto | 22 import "tensorflow/core/protobuf/device_filters.proto";
|
| D | device_filters.proto | 59 repeated string device_filters = 1; field
|
| D | master.proto | 248 repeated string device_filters = 2; field
|
| D | config.proto | 434 repeated string device_filters = 4; field
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | context.cc | 834 const string& remote_worker, const std::vector<string>& device_filters) { in SetRemoteDeviceFilters() argument
852 for (auto& filter : device_filters) { in SetRemoteDeviceFilters()
863 for (auto& filter : device_filters) { in SetRemoteDeviceFilters()
|
| D | context.h | 368 const std::vector<string>& device_filters);
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | config_test.py | 676 device_filters=['/job:worker/task:1'])
684 self.assertEqual(['/job:worker/task:1'], new_config.device_filters)
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api.cc | 778 std::vector<string> device_filters(tdf.second.device_filters_size()); in TFE_ContextSetServerDef()
780 device_filters[i] = tdf.second.device_filters(i); in TFE_ContextSetServerDef()
784 ctx->context->SetRemoteDeviceFilters(remote_worker, device_filters); in TFE_ContextSetServerDef()
|
| /external/tensorflow/tensorflow/core/ |
| D | BUILD | 189 "protobuf/device_filters.proto",
|
| /external/tensorflow/ |
| D | RELEASE.md | 1471 * `RunConfig` now sets device_filters to restrict how workers and PS can communicate. This can spee…
|