Searched refs:curr_op (Results 1 – 5 of 5) sorted by relevance
34 auto curr_op = op_it->get(); in Run() local35 if (curr_op->type != OperatorType::kLstmCell) { in Run()39 const auto* curr_lstm_op = static_cast<LstmCellOperator*>(curr_op); in Run()49 *model, curr_op->inputs[LstmCellOperator::WEIGHTS_INPUT]) || in Run()51 *model, curr_op->inputs[LstmCellOperator::BIASES_INPUT])) { in Run()56 if (!model->GetArray(curr_op->outputs[LstmCellOperator::ACTIV_OUTPUT]) in Run()65 int num_input = model->GetArray(curr_op->inputs[LstmCellOperator::DATA_INPUT]) in Run()71 model->GetArray(curr_op->outputs[LstmCellOperator::ACTIV_OUTPUT]) in Run()78 curr_op->inputs[LstmCellOperator::ACTIV_OUTPUT]; in Run()82 curr_op->inputs[LstmCellOperator::PREV_ACTIV_INPUT]; in Run()[all …]
142 enum mv_op curr_op; in mv_seq_exec() local162 curr_op = get_cfg_seq_op(&seq_arr[seq_idx]); in mv_seq_exec()163 op_execute_func_arr[curr_op](serdes_num, &seq_arr[seq_idx], in mv_seq_exec()
304 curr_op = op.op305 curr_op_type = set([curr_op.type])308 curr_op = curr_op.inputs[0].op309 curr_op_type = set([curr_op.type])315 curr_op.name.find('delayed_quant/Merge') > 0):320 for input_op in curr_op.inputs:
179 grpc_transport_stream_op_batch* curr_op; member805 static bool op_can_be_run(grpc_transport_stream_op_batch* curr_op, in op_can_be_run() argument932 else if (curr_op->send_initial_metadata && in op_can_be_run()936 } else if (curr_op->send_message && in op_can_be_run()940 } else if (curr_op->send_message && in op_can_be_run()944 } else if (curr_op->send_trailing_metadata && in op_can_be_run()948 } else if (curr_op->recv_initial_metadata && in op_can_be_run()952 } else if (curr_op->recv_message && in op_can_be_run()956 } else if (curr_op->cancel_stream && in op_can_be_run()960 } else if (curr_op->recv_trailing_metadata) { in op_can_be_run()[all …]
433 curr_op = queue.popleft()434 if curr_op in visited: continue435 visited.add(curr_op)436 if curr_op in from_ops:437 target_op = curr_op439 queue.extend(t.op for t in _NonEagerInputs(curr_op, xs))