Home
last modified time | relevance | path

Searched refs:TaskType (Results 1 – 23 of 23) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/batching_util/
Dshared_batch_scheduler.h42 template <typename TaskType>
101 template <typename TaskType>
103 : public std::enable_shared_from_this<SharedBatchScheduler<TaskType>> {
122 std::shared_ptr<SharedBatchScheduler<TaskType>>* scheduler);
163 std::function<void(std::unique_ptr<Batch<TaskType>>)>
165 std::unique_ptr<BatchScheduler<TaskType>>* queue);
183 using QueueList = std::list<std::unique_ptr<internal::Queue<TaskType>>>;
227 template <typename TaskType>
231 std::function<void(std::unique_ptr<Batch<TaskType>>)>;
233 Queue(const typename SharedBatchScheduler<TaskType>::QueueOptions& options,
[all …]
Dbatch_scheduler.h77 template <typename TaskType>
86 void AddTask(std::unique_ptr<TaskType> task);
90 std::unique_ptr<TaskType> RemoveTask();
99 const TaskType& task(int i) const;
102 TaskType* mutable_task(int i);
120 std::vector<std::unique_ptr<TaskType>> tasks_ GUARDED_BY(mu_);
136 template <typename TaskType>
159 virtual Status Schedule(std::unique_ptr<TaskType>* task) = 0;
190 template <typename TaskType>
191 Batch<TaskType>::~Batch() { in ~Batch()
[all …]
Dadaptive_shared_batch_scheduler.h40 template <typename TaskType>
43 template <typename TaskType>
73 template <typename TaskType>
76 AdaptiveSharedBatchScheduler<TaskType>> {
121 std::shared_ptr<AdaptiveSharedBatchScheduler<TaskType>>* scheduler);
134 using BatchProcessor = std::function<void(std::unique_ptr<Batch<TaskType>>)>;
139 std::unique_ptr<BatchScheduler<TaskType>>* queue);
148 friend class internal::ASBSQueue<TaskType>;
153 void CallbackWrapper(const internal::ASBSBatch<TaskType>* batch,
167 void AddBatch(const internal::ASBSBatch<TaskType>* batch,
[all …]
Dserial_device_batch_scheduler.h37 template <typename TaskType>
40 template <typename TaskType>
66 template <typename TaskType>
68 SerialDeviceBatchScheduler<TaskType>> {
104 std::shared_ptr<SerialDeviceBatchScheduler<TaskType>>* scheduler);
113 using BatchProcessor = std::function<void(std::unique_ptr<Batch<TaskType>>)>;
118 std::unique_ptr<BatchScheduler<TaskType>>* queue);
132 friend class internal::SDBSQueue<TaskType>;
140 void AddBatch(const internal::SDBSBatch<TaskType>* batch);
143 void RemoveQueue(const internal::SDBSQueue<TaskType>* queue);
[all …]
Dbasic_batch_scheduler.h136 template <typename TaskType>
137 class BasicBatchScheduler : public BatchScheduler<TaskType> {
185 std::function<void(std::unique_ptr<Batch<TaskType>>)>
191 Status Schedule(std::unique_ptr<TaskType>* task) override;
201 std::unique_ptr<BatchScheduler<TaskType>> shared_scheduler_queue);
205 std::unique_ptr<BatchScheduler<TaskType>> shared_scheduler_queue_;
213 template <typename TaskType>
214 Status BasicBatchScheduler<TaskType>::Create( in Create()
216 std::function<void(std::unique_ptr<Batch<TaskType>>)> in Create()
219 typename SharedBatchScheduler<TaskType>::Options shared_scheduler_options; in Create()
[all …]
/external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/
Drun_config_test.py63 run_config_lib.TaskType.PS: ["host1:1", "host2:2"],
64 run_config_lib.TaskType.WORKER: ["host3:3", "host4:4", "host5:5"]
67 "type": run_config_lib.TaskType.WORKER,
79 self.assertEqual(config.task_type, run_config_lib.TaskType.WORKER)
85 run_config_lib.TaskType.PS: ["localhost:9990"],
91 "type": run_config_lib.TaskType.WORKER,
104 self.assertEqual(config.task_type, run_config_lib.TaskType.WORKER)
109 tf_config = {"cluster": {run_config_lib.TaskType.WORKER: ["host1:1"]}}
117 run_config_lib.TaskType.PS: ["host1:1", "host2:2"],
118 run_config_lib.TaskType.WORKER: ["host3:3", "host4:4", "host5:5"]
[all …]
Drun_config.py65 class TaskType(object): class
159 self._is_chief = (self._task_type == TaskType.MASTER and
163 self._is_chief = (self._task_type == TaskType.WORKER and
D__init__.py337 from tensorflow.contrib.learn.python.learn.estimators.run_config import TaskType
Dlinear_test.py490 run_config.TaskType.PS: ['fake_ps_0', 'fake_ps_1']
1006 run_config.TaskType.PS: ['fake_ps_0', 'fake_ps_1']
1461 run_config.TaskType.PS: ['fake_ps_0', 'fake_ps_1']
1635 run_config.TaskType.PS: ['fake_ps_0', 'fake_ps_1']
Destimator_test.py1331 tf_config = {'cluster': {run_config.TaskType.PS: ['fake_ps_0']}}
1360 tf_config = {'cluster': {run_config.TaskType.PS: ['fake_ps_0']}}
1387 run_config.TaskType.PS: ['fake_ps_0']
1390 'type': run_config.TaskType.WORKER,
Ddnn_test.py945 run_config.TaskType.PS: ['fake_ps_0', 'fake_ps_1']
1511 run_config.TaskType.PS: ['fake_ps_0', 'fake_ps_1']
Ddnn_linear_combined_test.py497 run_config.TaskType.PS: ['fake_ps_0', 'fake_ps_1']
1642 run_config.TaskType.PS: ['fake_ps_0', 'fake_ps_1']
/external/tensorflow/tensorflow/python/tpu/
Dtpu_config_test.py49 run_config_lib.TaskType.CHIEF: ['host3:3'],
50 run_config_lib.TaskType.WORKER: ['host3:4']
53 'type': run_config_lib.TaskType.CHIEF,
64 run_config_lib.TaskType.CHIEF: ['host3:3'],
65 run_config_lib.TaskType.WORKER: ['host3:4']
68 'type': run_config_lib.TaskType.CHIEF,
134 run_config_lib.TaskType.CHIEF: ['host0:0'],
137 'type': run_config_lib.TaskType.EVALUATOR,
Dtpu_config.py228 self.task_type != run_config_lib.TaskType.EVALUATOR):
/external/tensorflow/tensorflow/contrib/learn/python/learn/
Dlearn_runner_test.py131 run_config_lib.TaskType.PS: ["localhost:1234", "localhost:1235"],
132 run_config_lib.TaskType.WORKER: ["localhost:1236", "localhost:1237"],
133 run_config_lib.TaskType.MASTER: ["localhost:1238"],
336 "type": run_config_lib.TaskType.WORKER
351 "type": run_config_lib.TaskType.MASTER
366 "type": run_config_lib.TaskType.PS
Dlearn_runner.py288 if config.task_type == run_config_lib.TaskType.MASTER:
292 elif config.task_type == run_config_lib.TaskType.PS:
294 elif config.task_type == run_config_lib.TaskType.WORKER:
Dexperiment_test.py196 run_config_lib.TaskType.PS: ['host1:2222', 'host2:2222'],
197 run_config_lib.TaskType.WORKER:
287 'type': run_config_lib.TaskType.WORKER,
317 job_name=run_config_lib.TaskType.WORKER,
836 'type': run_config_lib.TaskType.PS,
/external/tensorflow/tensorflow/contrib/factorization/python/ops/
Dwals_test.py404 run_config_lib.TaskType.PS: ['host1:1', 'host2:2'],
405 run_config_lib.TaskType.WORKER: ['host3:3', 'host4:4']
408 'type': run_config_lib.TaskType.WORKER,
/external/tensorflow/tensorflow/contrib/distribute/python/
Dmulti_worker_test_base.py302 for task_type in [run_config.TaskType.CHIEF, run_config.TaskType.WORKER]:
Dparameter_server_strategy_test.py58 CHIEF = run_config.TaskType.CHIEF
59 WORKER = run_config.TaskType.WORKER
60 PS = run_config.TaskType.PS
/external/python/cpython3/Modules/
D_asynciomodule.c107 static PyTypeObject TaskType; variable
112 #define Task_CheckExact(obj) (Py_TYPE(obj) == &TaskType)
115 #define Task_Check(obj) PyObject_TypeCheck(obj, &TaskType)
2409 static PyTypeObject TaskType = { variable
3348 if (PyType_Ready(&TaskType) < 0) { in PyInit__asyncio()
3366 Py_INCREF(&TaskType); in PyInit__asyncio()
3367 if (PyModule_AddObject(m, "Task", (PyObject *)&TaskType) < 0) { in PyInit__asyncio()
3368 Py_DECREF(&TaskType); in PyInit__asyncio()
/external/gemmlowp/internal/
Dmulti_thread_gemm.h678 TaskType; in MultiThreadGemm() typedef
680 new TaskType(context, kernel, lhs_block, packed_rhs, result, in MultiThreadGemm()
/external/libchrome/base/message_loop/
Dmessage_loop_unittest.cc121 enum TaskType { enum
135 TaskItem(TaskType t, int c, bool s) in TaskItem()
141 TaskType type;
150 std::ostream& operator <<(std::ostream& os, TaskType type) { in operator <<()
177 void RecordStart(TaskType type, int cookie) { in RecordStart()
183 void RecordEnd(TaskType type, int cookie) { in RecordEnd()