Lines Matching refs:task_num
63 int task_num = job->args->task_number; in rknn_get_task_number() local
66 task_num = job->args->subcore_task[core_index].task_number; in rknn_get_task_number()
68 task_num = job->args->subcore_task[core_index + 2].task_number; in rknn_get_task_number()
70 return task_num; in rknn_get_task_number()
180 subcore_data->task_num -= rknn_get_task_number(job, core_index); in rknpu_job_wait()
385 subcore_data->task_num -= rknn_get_task_number(job, core_index); in rknpu_job_done()
421 if (rknpu_dev->subcore_datas[0].task_num > in rknpu_job_schedule()
422 rknpu_dev->subcore_datas[1].task_num) { in rknpu_job_schedule()
427 if (rknpu_dev->subcore_datas[task_num_list[0]].task_num > in rknpu_job_schedule()
428 rknpu_dev->subcore_datas[2].task_num) { in rknpu_job_schedule()
433 } else if (rknpu_dev->subcore_datas[task_num_list[1]].task_num > in rknpu_job_schedule()
434 rknpu_dev->subcore_datas[2].task_num) { in rknpu_job_schedule()
459 subcore_data->task_num += rknn_get_task_number(job, i); in rknpu_job_schedule()
486 subcore_data->task_num -= in rknpu_job_abort()