Lines Matching refs:task_id

6719 -int LstmInputMulWeightRun(void *cdata, int task_id, float, float) {
6722 -  kernel->InputWeightMatMul(task_id);
6726 -int LstmSequenceLoopRun(void *cdata, int task_id, float, float) {
6729 -  auto ret = kernel->DoSequenceLoop(task_id);
7138 -void LstmCPUKernel::InputWeightMatMul(int task_id) const {
7139 -  int current_start_oc = task_id * input_thread_stride_ * col_tile_;
7154 -int LstmCPUKernel::DoSequenceLoop(int task_id) {
7155 -  if (task_id == 0) {
7159 -  if (task_id == 1) {
7436 +int LstmSequenceLoopRun(void *cdata, int task_id, float, float) {
7439 +  auto ret = kernel->DoSequenceLoop(task_id);
7730 +  auto MatMulCoreFunc = [this, &weight, &bias, &gate, &stride](void *, int task_id, float, float) {
7731 +    int current_start_oc = task_id * stride * col_tile_;
7758 +int LstmFp32BaseCPUKernel::DoSequenceLoop(int task_id) {
7759 +  if (task_id == 0) {
7763 +  if (task_id == 1) {
7843 +  int DoSequenceLoop(int task_id);