| /external/tensorflow/tensorflow/python/training/ |
| D | server_lib.py | 31 def _make_server_def(server_or_cluster_def, job_name, task_index, protocol, argument
62 if task_index is not None:
63 server_def.task_index = task_index
79 if task_index is None:
82 task_index = task_indices[0]
91 task_index=task_index,
114 task_index=None, argument
147 task_index, protocol, config)
418 def task_address(self, job_name, task_index): argument
437 return job[task_index]
[all …]
|
| D | server_lib_test.py | 200 task_index=0,
268 }, job_name="local", task_index=0)
400 cluster=cluster_def, job_name="local", task_index=0, protocol="grpc")
418 cluster=cluster_def, job_name="local", task_index=1, protocol="grpc")
438 cluster=cluster_def, job_name="worker", task_index=2, protocol="grpc")
464 cluster=cluster_def, job_name="worker", task_index=2, protocol="grpc")
|
| /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 | 135 auto iter = job.tasks().find(server_def.task_index()); in GetHostAndPort()
137 return errors::Internal("Task ", server_def.task_index(), in GetHostAndPort()
188 "/task:", server_def_.task_index()); in Init()
348 if (job.name() == *options.job_name && task.first == options.task_index) { in ParseChannelSpec()
377 "/task:", options.task_index); in WorkerCacheFactory()
|
| /external/tensorflow/tensorflow/core/protobuf/ |
| D | cluster.proto | 38 // cluster { $CLUSTER } job_name: 'local' task_index: 0
47 // cluster { $CLUSTER } job_name: 'local' task_index: 0
48 // cluster { $CLUSTER } job_name: 'local' task_index: 1
61 // cluster { $CLUSTER } job_name: 'worker' task_index: 0
62 // cluster { $CLUSTER } job_name: 'worker' task_index: 1
63 // cluster { $CLUSTER } job_name: 'worker' task_index: 2
64 // cluster { $CLUSTER } job_name: 'ps' task_index: 0
65 // cluster { $CLUSTER } job_name: 'ps' task_index: 1
|
| /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/libchrome/base/task_scheduler/ |
| D | test_task_factory.cc | 50 void TestTaskFactory::RunTaskCallback(size_t task_index, in RunTaskCallback() argument
82 DCHECK_LE(task_index, num_posted_tasks_); in RunTaskCallback()
86 task_index != ran_tasks_.size()) { in RunTaskCallback()
93 if (ran_tasks_.find(task_index) != ran_tasks_.end()) in RunTaskCallback()
95 ran_tasks_.insert(task_index); in RunTaskCallback()
|
| D | test_task_factory.h | 66 void RunTaskCallback(size_t task_index,
|
| /external/tensorflow/tensorflow/python/keras/benchmarks/ |
| D | distribution_util.py | 141 def configure_cluster(worker_hosts=None, task_index=-1): argument
158 if num_workers > 1 and task_index < 0:
160 task_index = 0 if num_workers == 1 else task_index
167 "index": task_index
|
| /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/core/distributed_runtime/ |
| D | master_env.h | 40 int task_index; member
52 task_index = server_def.task_index(); in WorkerCacheFactoryOptions()
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | remote_execution_test.py | 47 task_index): argument
60 task_index=task_index,
86 task_index=0))
187 task_index=0))
202 task_index=0))
|
| D | remote.py | 83 task_index=0, argument
175 task_index=task_index,
|
| D | remote_cluster_test.py | 47 task_index): argument
60 task_index=task_index,
84 task_index=0)
91 task_index=0)
98 task_index=0)
105 task_index=0)
113 task_index=0)
121 task_index=0)
140 task_index=0))
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/metal/ |
| D | inference_context.cc | 72 void AddUsage(ValueId id, int task_index, in AddUsage() argument
77 (*usage_records)[id].x = task_index; in AddUsage()
78 (*usage_records)[id].y = task_index; in AddUsage()
81 (*usage_records)[id].y = task_index; in AddUsage()
661 for (auto& task_index : task_ids_with_preallocated_tensors_) { in UpdatePreallocatedTensors() local
662 auto& task = nodes_[task_index].task; in UpdatePreallocatedTensors()
663 const auto& src_ids = nodes_[task_index].inputs; in UpdatePreallocatedTensors()
670 const auto& dst_ids = nodes_[task_index].outputs; in UpdatePreallocatedTensors()
|
| /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…
|
| 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\', \…
|
| /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.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\', \…
|
| /external/tensorflow/tensorflow/python/distribute/coordinator/ |
| D | utils.py | 40 task_index=cluster_resolver.task_id,
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | multi_worker_test_base.py | 122 task_index=i,
131 task_index=i,
140 task_index=0,
149 task_index=0,
255 task_index=task_id,
|