/external/tensorflow/tensorflow/core/kernels/batching_util/ |
D | shared_batch_scheduler.h | 42 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 …]
|
D | batch_scheduler.h | 77 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 …]
|
D | adaptive_shared_batch_scheduler.h | 40 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 …]
|
D | serial_device_batch_scheduler.h | 37 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 …]
|
D | basic_batch_scheduler.h | 136 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/ |
D | run_config_test.py | 63 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 …]
|
D | run_config.py | 65 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__.py | 337 from tensorflow.contrib.learn.python.learn.estimators.run_config import TaskType
|
D | linear_test.py | 490 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']
|
D | estimator_test.py | 1331 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,
|
D | dnn_test.py | 945 run_config.TaskType.PS: ['fake_ps_0', 'fake_ps_1'] 1511 run_config.TaskType.PS: ['fake_ps_0', 'fake_ps_1']
|
D | dnn_linear_combined_test.py | 497 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/ |
D | tpu_config_test.py | 49 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,
|
D | tpu_config.py | 228 self.task_type != run_config_lib.TaskType.EVALUATOR):
|
/external/tensorflow/tensorflow/contrib/learn/python/learn/ |
D | learn_runner_test.py | 131 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
|
D | learn_runner.py | 288 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:
|
D | experiment_test.py | 196 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/ |
D | wals_test.py | 404 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/ |
D | multi_worker_test_base.py | 302 for task_type in [run_config.TaskType.CHIEF, run_config.TaskType.WORKER]:
|
D | parameter_server_strategy_test.py | 58 CHIEF = run_config.TaskType.CHIEF 59 WORKER = run_config.TaskType.WORKER 60 PS = run_config.TaskType.PS
|
/external/python/cpython3/Modules/ |
D | _asynciomodule.c | 107 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/ |
D | multi_thread_gemm.h | 678 TaskType; in MultiThreadGemm() typedef 680 new TaskType(context, kernel, lhs_block, packed_rhs, result, in MultiThreadGemm()
|
/external/libchrome/base/message_loop/ |
D | message_loop_unittest.cc | 121 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()
|