| /external/tensorflow/tensorflow/core/protobuf/ |
| tensorflow_server.proto | 42 int32 task_index = 3;
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| master_env.h | 40 int task_index; member in struct:tensorflow::WorkerCacheFactoryOptions
52 task_index = server_def.task_index();
|
| session_mgr.cc | 49 server_def.task_index());
|
| master.cc | 380 // Set the server_def's job_name and task_index fields.
413 worker_cache_factory_options.task_index = task.first;
434 name.task == worker_cache_factory_options.task_index &&
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| grpc_testlib_server.cc | 39 int num_cpus, int num_gpus, int task_index,
43 options->set_task_index(task_index);
84 int task_index = 0; local
88 tensorflow::Flag("tf_task", &task_index, "task index"),
101 num_gpus, task_index, &def);
|
| grpc_tensorflow_server.cc | 42 int task_index, ServerDef* options) {
45 options->set_task_index(task_index);
76 if (options->task_index() >= my_num_tasks) {
77 return errors::InvalidArgument("Task index ", options->task_index(),
101 int task_index = 0; local
105 tensorflow::Flag("task_id", &task_index, "task id"),
117 task_index, &server_def);
|
| grpc_server_lib.cc | 122 // otherwise if 'task_index=-1' the program will abort.
128 auto iter = job.tasks().find(server_def_.task_index());
130 return errors::InvalidArgument("Task ", server_def_.task_index(),
156 "/task:", server_def_.task_index());
294 if (job.name() == *options.job_name && task.first == options.task_index) {
323 "/task:", options.task_index);
|
| /external/tensorflow/tensorflow/python/training/ |
| server_lib.py | 30 def _make_server_def(server_or_cluster_def, job_name, task_index, protocol,
42 task_index: (Optional.) Specifies the task index of the server in its job.
63 if task_index is not None:
64 server_def.task_index = task_index
80 if task_index is None:
83 task_index = task_indices[0]
85 raise ValueError("Must specify an explicit `task_index`.")
91 job_name=job_name, task_index=task_index, protocol=protocol
[all...] |
| server_lib_test.py | 201 task_index=0,
212 }, job_name="local", task_index=0)
348 cluster=cluster_def, job_name="local", task_index=0, protocol="grpc")
354 job_name: 'local' task_index: 0 protocol: 'grpc'
366 cluster=cluster_def, job_name="local", task_index=1, protocol="grpc")
373 job_name: 'local' task_index: 1 protocol: 'grpc'
386 cluster=cluster_def, job_name="worker", task_index=2, protocol="grpc")
396 job_name: 'worker' task_index: 2 protocol: 'grpc'
412 cluster=cluster_def, job_name="worker", task_index=2, protocol="grpc")
421 job_name: 'worker' task_index: 2 protocol: 'grpc
[all...] |
| /external/toolchain-utils/bestflags/ |
| steering_test.py | 81 task_index = 0
90 test_ranges = range(task_index, task_index + NUMBER_OF_TASKS)
100 task_index += NUMBER_OF_TASKS
|
| /external/libchrome/base/message_loop/ |
| message_loop_unittest.cc | 1371 int task_index = 0; local
1403 int task_index = 0; local
1433 int task_index = 0; local
1470 int task_index = 0; local
1501 int task_index = 0; local
1555 int task_index = 0; local
1616 int task_index = 0; local
1651 int task_index = 0; local
[all...] |
| /external/tensorflow/tensorflow/contrib/eager/python/ |
| remote_test.py | 61 task_index):
74 task_index=task_index,
101 task_index=0))
176 task_index=0))
191 task_index=0))
|
| /external/tensorflow/tensorflow/python/eager/ |
| remote.py | 75 task_index=0,
|
| /external/tensorflow/tensorflow/contrib/opt/python/training/ |
| agn_optimizer.py | 215 def get_init_op(self, task_index):
234 def make_session_run_hook(self, is_chief, task_index):
236 return _AGNOptimizerHook(self, is_chief, task_index)
241 def __init__(self, agn_optimizer, is_chief, task_index):
247 task_index: int, task_index of worker
251 self._task_index = task_index
|
| model_average_optimizer_test.py | 47 cs, job_name="worker", protocol=protocol, task_index=ix, start=True)
52 cs, job_name="ps", protocol=protocol, task_index=ix, start=True)
|
| agn_optimizer_test.py | 50 cs, job_name="worker", protocol=protocol, task_index=ix, start=True)
55 cs, job_name="ps", protocol=protocol, task_index=ix, start=True)
|
| drop_stale_gradient_optimizer_test.py | 43 cs, job_name='worker', task_index=ix, start=True)
46 server_lib.Server(cs, job_name='ps', task_index=0, start=True)
|
| elastic_average_optimizer_test.py | 53 cs, job_name="worker", protocol=protocol, task_index=ix, start=True)
58 cs, job_name="ps", protocol=protocol, task_index=ix, start=True)
|
| /external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/ |
| run_config.py | 217 task_index = task_env.get('index')
218 return int(task_index) if task_index else 0
|
| /external/tensorflow/tensorflow/contrib/distribute/python/ |
| multi_worker_test_base.py | 99 task_index=i,
108 task_index=i,
117 task_index=0,
126 task_index=0,
|
| /external/tensorflow/tensorflow/python/debug/lib/ |
| grpc_tensorflow_server.py | 106 server_def.task_index = FLAGS.task_id
|
| /external/tensorflow/tensorflow/contrib/gdr/ |
| gdr_server_lib.cc | 35 auto iter = job.tasks().find(server_def.task_index());
|
| /external/tensorflow/tensorflow/contrib/verbs/ |
| verbs_server_lib.cc | 55 "/task:", server_def.task_index());
|
| /external/tensorflow/tensorflow/contrib/mpi/ |
| README.md | 9 ```server = tf.train.Server(cluster, job_name="local", task_index=0, protocol='grpc+mpi') # default protocol is 'grpc'```
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| eager_service_impl.cc | 98 request->server_def().task_index()),
|
