| /external/tensorflow/tensorflow/python/training/ |
| D | server_lib.py | 30 def _make_server_def(server_or_cluster_def, job_name, task_index, protocol, argument
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]
91 job_name=job_name, task_index=task_index, protocol=protocol)
113 task_index=None, argument
147 job_name, task_index, protocol, config)
409 def task_address(self, job_name, task_index): argument
428 return job[task_index]
[all …]
|
| D | 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")
366 cluster=cluster_def, job_name="local", task_index=1, protocol="grpc")
386 cluster=cluster_def, job_name="worker", task_index=2, protocol="grpc")
412 cluster=cluster_def, job_name="worker", task_index=2, protocol="grpc")
|
| /external/libchrome/base/message_loop/ |
| D | message_loop_unittest.cc | 1371 int task_index = 0; in TEST_P() local
1372 EXPECT_EQ(order.Get(task_index++), TaskItem(RUNS, 1, true)); in TEST_P()
1373 EXPECT_EQ(order.Get(task_index++), TaskItem(ORDERED, 2, true)); in TEST_P()
1374 EXPECT_EQ(order.Get(task_index++), TaskItem(ORDERED, 2, false)); in TEST_P()
1375 EXPECT_EQ(order.Get(task_index++), TaskItem(RUNS, 1, false)); in TEST_P()
1376 EXPECT_EQ(order.Get(task_index++), TaskItem(ORDERED, 3, true)); in TEST_P()
1377 EXPECT_EQ(order.Get(task_index++), TaskItem(ORDERED, 3, false)); in TEST_P()
1378 EXPECT_EQ(static_cast<size_t>(task_index), order.Size()); in TEST_P()
1403 int task_index = 0; in TEST_P() local
1404 EXPECT_EQ(order.Get(task_index++), TaskItem(RUNS, 1, true)); in TEST_P()
[all …]
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | grpc_tensorflow_server.cc | 42 int task_index, ServerDef* options) { in FillServerDef() argument
45 options->set_task_index(task_index); in FillServerDef()
76 if (options->task_index() >= my_num_tasks) { in FillServerDef()
77 return errors::InvalidArgument("Task index ", options->task_index(), in FillServerDef()
101 int task_index = 0; in main() local
105 tensorflow::Flag("task_id", &task_index, "task id"), in main()
117 task_index, &server_def); in main()
|
| D | grpc_testlib_server.cc | 39 int num_cpus, int num_gpus, int task_index, in FillServerDef() argument
43 options->set_task_index(task_index); in FillServerDef()
84 int task_index = 0; in main() local
88 tensorflow::Flag("tf_task", &task_index, "task index"), in main()
101 num_gpus, task_index, &def); in main()
|
| D | grpc_server_lib.cc | 128 auto iter = job.tasks().find(server_def_.task_index()); in Init()
130 return errors::InvalidArgument("Task ", server_def_.task_index(), in Init()
156 "/task:", server_def_.task_index()); in Init()
294 if (job.name() == *options.job_name && task.first == options.task_index) { in ParseChannelSpec()
323 "/task:", options.task_index); in WorkerCacheFactory()
|
| /external/tensorflow/tensorflow/core/protobuf/ |
| D | cluster.proto | 37 // cluster { $CLUSTER } job_name: 'local' task_index: 0
46 // cluster { $CLUSTER } job_name: 'local' task_index: 0
47 // cluster { $CLUSTER } job_name: 'local' task_index: 1
60 // cluster { $CLUSTER } job_name: 'worker' task_index: 0
61 // cluster { $CLUSTER } job_name: 'worker' task_index: 1
62 // cluster { $CLUSTER } job_name: 'worker' task_index: 2
63 // cluster { $CLUSTER } job_name: 'ps' task_index: 0
64 // cluster { $CLUSTER } job_name: 'ps' task_index: 1
|
| /external/toolchain-utils/bestflags/ |
| D | 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/tensorflow/tensorflow/contrib/eager/python/ |
| D | remote_test.py | 61 task_index): argument
74 task_index=task_index,
101 task_index=0))
176 task_index=0))
191 task_index=0))
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | master_env.h | 40 int task_index; member
52 task_index = server_def.task_index(); in WorkerCacheFactoryOptions()
|
| /external/tensorflow/tensorflow/contrib/opt/python/training/ |
| D | elastic_average_optimizer.py | 351 def get_init_op(self, task_index): argument
369 if i == task_index:
374 sync_queues[task_index].dequeue_many(len(sync_queues) - 1))
396 def make_session_run_hook(self, is_chief, task_index): argument
398 return _ElasticAverageOptimizerHook(self, is_chief, task_index)
448 def __init__(self, ea_optimizer, is_chief, task_index): argument
457 self._task_index = task_index
|
| D | agn_optimizer.py | 215 def get_init_op(self, task_index): argument
234 def make_session_run_hook(self, is_chief, task_index): argument
236 return _AGNOptimizerHook(self, is_chief, task_index)
241 def __init__(self, agn_optimizer, is_chief, task_index): argument
251 self._task_index = task_index
|
| D | 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)
|
| D | 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)
|
| /external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/ |
| D | 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/ |
| D | multi_worker_test_base.py | 99 task_index=i,
108 task_index=i,
117 task_index=0,
126 task_index=0,
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.train.-server-def.pbtxt | 19 name: "task_index"
|
| D | tensorflow.train.-server.pbtxt | 15 …argspec: "args=[\'self\', \'server_or_cluster_def\', \'job_name\', \'task_index\', \'protocol\', \…
|
| D | tensorflow.distribute.-server.pbtxt | 15 …argspec: "args=[\'self\', \'server_or_cluster_def\', \'job_name\', \'task_index\', \'protocol\', \…
|
| D | tensorflow.train.-cluster-spec.pbtxt | 31 …argspec: "args=[\'self\', \'job_name\', \'task_index\'], varargs=None, keywords=None, defaults=Non…
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.train.-server-def.pbtxt | 19 name: "task_index"
|
| D | tensorflow.distribute.-server.pbtxt | 15 …argspec: "args=[\'self\', \'server_or_cluster_def\', \'job_name\', \'task_index\', \'protocol\', \…
|
| D | tensorflow.train.-cluster-spec.pbtxt | 31 …argspec: "args=[\'self\', \'job_name\', \'task_index\'], varargs=None, keywords=None, defaults=Non…
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | remote.py | 75 task_index=0,
|
| /external/tensorflow/tensorflow/python/debug/lib/ |
| D | grpc_tensorflow_server.py | 106 server_def.task_index = FLAGS.task_id
|