Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/batching_util/
Dbatch_scheduler.h79 template <typename TaskType>
89 void AddTask(std::unique_ptr<TaskType> task);
93 std::unique_ptr<TaskType> RemoveTask();
102 const TaskType& task(int i) const;
105 TaskType* mutable_task(int i);
126 std::vector<std::unique_ptr<TaskType>> tasks_ TF_GUARDED_BY(mu_);
147 template <typename TaskType>
170 virtual Status Schedule(std::unique_ptr<TaskType>* task) = 0;
201 template <typename TaskType>
202 Batch<TaskType>::Batch() : Batch(0) {} in Batch()
[all …]
Dshared_batch_scheduler.h48 template <typename TaskType>
107 template <typename TaskType>
109 : public std::enable_shared_from_this<SharedBatchScheduler<TaskType>> {
128 std::shared_ptr<SharedBatchScheduler<TaskType>>* scheduler);
186 std::function<Status(std::unique_ptr<TaskType>* input_task,
188 std::vector<std::unique_ptr<TaskType>>* output_tasks)>
200 std::function<void(std::unique_ptr<Batch<TaskType>>)>
202 std::unique_ptr<BatchScheduler<TaskType>>* queue);
220 using QueueList = std::list<std::unique_ptr<internal::Queue<TaskType>>>;
264 template <typename TaskType>
[all …]
Dadaptive_shared_batch_scheduler.h45 template <typename TaskType>
48 template <typename TaskType>
78 template <typename TaskType>
81 AdaptiveSharedBatchScheduler<TaskType>> {
130 std::shared_ptr<AdaptiveSharedBatchScheduler<TaskType>>* scheduler);
154 std::function<Status(std::unique_ptr<TaskType>* input_task, int first_size,
156 std::vector<std::unique_ptr<TaskType>>* output_tasks)>
160 using BatchProcessor = std::function<void(std::unique_ptr<Batch<TaskType>>)>;
165 std::unique_ptr<BatchScheduler<TaskType>>* queue);
174 friend class internal::ASBSQueue<TaskType>;
[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.h137 template <typename TaskType>
138 class BasicBatchScheduler : public BatchScheduler<TaskType> {
173 std::shared_ptr<SharedBatchScheduler<TaskType>> shared_batch_scheduler =
243 std::function<Status(std::unique_ptr<TaskType>* input_task,
245 std::vector<std::unique_ptr<TaskType>>* output_tasks)>
268 std::function<void(std::unique_ptr<Batch<TaskType>>)>
274 Status Schedule(std::unique_ptr<TaskType>* task) override;
284 std::unique_ptr<BatchScheduler<TaskType>> shared_scheduler_queue);
288 std::unique_ptr<BatchScheduler<TaskType>> shared_scheduler_queue_;
296 template <typename TaskType>
[all …]
/external/tensorflow/tensorflow/lite/kernels/
Dcpu_backend_threadpool.h36 template <typename TaskType>
37 void Execute(int tasks_count, TaskType* tasks, in Execute()
48 template <typename TaskType>
49 void Execute(int tasks_count, TaskType* tasks,
/external/ruy/ruy/
Dthread_pool.h75 template <typename TaskType>
76 void Execute(int task_count, TaskType* tasks) { in Execute()
77 ExecuteImpl(task_count, sizeof(TaskType), static_cast<Task*>(tasks)); in Execute()
/external/llvm-project/openmp/tools/archer/
Dompt-tsan.cpp411 int TaskType{0}; member
445 : InBarrier(false), TaskType(taskType), BarrierIndex(0), RefCount(1), in TaskData()
457 : InBarrier(false), TaskType(taskType), BarrierIndex(0), RefCount(1), in TaskData()
466 bool isIncluded() { return TaskType & ompt_task_undeferred; } in isIncluded()
467 bool isUntied() { return TaskType & ompt_task_untied; } in isUntied()
468 bool isFinal() { return TaskType & ompt_task_final; } in isFinal()
469 bool isMergable() { return TaskType & ompt_task_mergeable; } in isMergable()
470 bool isMerged() { return TaskType & ompt_task_merged; } in isMerged()
472 bool isExplicit() { return TaskType & ompt_task_explicit; } in isExplicit()
473 bool isImplicit() { return TaskType & ompt_task_implicit; } in isImplicit()
[all …]
/external/gemmlowp/internal/
Dmulti_thread_gemm.h377 template <typename TaskType>
378 void Execute(int tasks_count, TaskType* tasks) { in Execute()
706 TaskType; in MultiThreadGemm() typedef
708 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()
/external/tensorflow/tensorflow/python/distribute/
Dparameter_server_strategy_test.py61 CHIEF = run_config.TaskType.CHIEF
62 WORKER = run_config.TaskType.WORKER
63 PS = run_config.TaskType.PS
/external/python/cpython3/Modules/
D_asynciomodule.c112 static PyTypeObject TaskType; variable
117 #define Task_CheckExact(obj) Py_IS_TYPE(obj, &TaskType)
120 #define Task_Check(obj) PyObject_TypeCheck(obj, &TaskType)
2530 static PyTypeObject TaskType = { variable
3468 if (PyModule_AddType(m, &TaskType) < 0) { in PyInit__asyncio()