Searched refs:last_task (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/memory_management/ |
| D | internal.h | 85 : last_task(task_id), object_id(obj_id) {} in QueueRecord()
89 return (last_task > other.last_task) ||
90 (last_task == other.last_task && object_id > other.object_id);
94 TaskId last_task; member
|
| D | equality_assignment.h | 51 objects_in_use.top().last_task < usage_records[i].first_task) { in EqualityAssignmentWithHash()
65 {usage_records[i].last_task, assignment->object_ids[i]}); in EqualityAssignmentWithHash()
72 {usage_records[i].last_task, assignment->object_ids[i]}); in EqualityAssignmentWithHash()
107 dealloc_task.push_back(usage_records[i].last_task); in EqualityAssignment()
112 dealloc_task[best_obj] = usage_records[i].last_task; in EqualityAssignment()
|
| D | greedy_in_order_assignment.h | 72 objects_in_use.top().last_task < usage_records[i].first_task) {
130 {usage_records[i].last_task, assignment->object_ids[i]});
139 {usage_records[i].last_task, assignment->object_ids[i]});
166 objects_in_use.top().last_task < usage_records[i].first_task) { in GreedyInOrderAssignmentMultidimensional()
196 {usage_records[i].last_task, assignment->object_ids[i]}); in GreedyInOrderAssignmentMultidimensional()
202 {usage_records[i].last_task, assignment->object_ids[i]}); in GreedyInOrderAssignmentMultidimensional()
|
| D | greedy_by_size_assignment.cc | 91 if (usage_records[allocated_id].last_task < rec->first_task || in GreedyBySizeAssignment()
92 usage_records[allocated_id].first_task > rec->last_task) { in GreedyBySizeAssignment()
245 if (usage_records[rec_id].last_task < in GreedyBySizeDistPriorityAssignment()
248 usage_records[rec_id].last_task; in GreedyBySizeDistPriorityAssignment()
249 } else if (usage_records[best_rec_id].last_task < in GreedyBySizeDistPriorityAssignment()
252 usage_records[best_rec_id].last_task; in GreedyBySizeDistPriorityAssignment()
|
| D | types.h | 38 TaskId last_task; member
41 : tensor_size(size), first_task(first), last_task(last) {} in TensorUsageRecord()
|
| D | greedy_by_breadth_assignment.cc | 102 if (it != schedule.end() && it->first_task <= rec.last_task) { in GreedyByBreadthAssignment()
109 if (it->last_task >= rec.first_task) { in GreedyByBreadthAssignment()
|
| D | internal.cc | 68 num_tasks = std::max(num_tasks, usage_records[i].last_task + 1); in CalculateTaskProfiles()
75 task_id <= usage_records[rec_id].last_task; ++task_id) { in CalculateTaskProfiles()
|
| D | min_cost_flow_assignment.cc | 50 objects_in_use.top().last_task < usage_records[i].first_task) { in Build()
54 objects_in_use.push({usage_records[i].last_task, i}); in Build()
|
| /external/tensorflow/tensorflow/core/kernels/batching_util/ |
| D | batch_resource_base.h | 149 const BatchResourceBase::BatchTask& last_task,
|
| D | batch_resource_base.cc | 560 auto& last_task = batch->task(batch->num_tasks() - 1); in ProcessFuncBatch() local
561 OpKernelContext* last_task_context = last_task.context; in ProcessFuncBatch()
616 last_task, args, &combined_outputs, [&](const Status& run_status) { in ProcessFuncBatch()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/gl/ |
| D | runtime.cc | 414 usage_rec->last_task = std::max(usage_rec->last_task, task_id); in UpdateUsageRecord()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | batch_kernels.cc | 154 const BatchTask& last_task, absl::Span<const Tensor> inputs, in ProcessFuncBatchImpl() argument
157 auto* last_task_context = last_task.context; in ProcessFuncBatchImpl()
|