Lines Matching refs:gate

3730      float16_t *gate = gate_buffer + row * col * i;
3731      LstmMatMulFp16(gate, input, weight_i, bias_i, row, deep, col, is_vec);
3779 -  float16_t *gate = buffer[1];
3781 +void LstmGateCompute(float16_t *gate, const float16_t *input, const float16_t *weight_i, const flo…
3787 +    float16_t *gate_loop = gate + lstm_param->seq_len_ * lstm_param->batch_ * lstm_param->hidden_s…
3793 +void LstmGateCompute(float16_t *gate, const float16_t *input, const float16_t *weight_i, const flo…
3798 …float16_t *gate_loop = gate + lstm_param->seq_len_ * lstm_param->batch_ * lstm_param->hidden_size_…
3812 +  float16_t *gate = buffer[1];
3813 +  LstmGateCompute(gate, packed_input, weight_i, input_bias, lstm_param);
3815    float16_t *input_gate = gate;
3816 …float16_t *forget_gate = gate + lstm_param->seq_len_ * lstm_param->batch_ * lstm_param->hidden_siz…
5869 -    MS_LOG(ERROR) << "LstmFp16CPUKernel malloc state gate buffer_ error.";
6172 +    MS_LOG(ERROR) << "LstmFp16CPUKernel malloc state gate buffer_ error.";
6369 + *    weight_bias: forth input, weight_ih + weight_hh + bias, the gate order is IFGO
6375 + *    weight_bias: forth input, weight_ih + weight_hh + proj + bias, the gate order is IFGO
7059 -  segment = gate_num * lstm_param_->batch_ * lstm_param_->hidden_size_;  // 2: state gate buffer
7182 -  float *gate = buffer[input_gate_index];
7183 -  float *input_gate = gate;
7184 -  float *forget_gate = gate + lstm_param_->seq_len_ * lstm_param_->batch_ * lstm_param_->hidden_si…
7185 -  float *cell_gate = gate + lstm_param_->seq_len_ * lstm_param_->batch_ * lstm_param_->hidden_size…
7186 -  float *output_gate = gate + lstm_param_->seq_len_ * lstm_param_->batch_ * lstm_param_->hidden_si…
7607 +  segment = kGateNum * lstm_param_->batch_ * lstm_param_->hidden_size_;  // 2: state gate buffer
7726 +  float *gate{nullptr};
7730 +  auto MatMulCoreFunc = [this, &weight, &bias, &gate, &stride](void *, int task_id, float, float) {
7740 +    auto c = gate + current_start_oc;
7749 +    gate = dst + lstm_param_->seq_len_ * lstm_param_->batch_ * lstm_param_->hidden_size_ * i;
8098 +  float *gate = buffer[kInputGateIndex];
8099 +  float *input_gate = gate;
8100 +  float *forget_gate = gate + lstm_param_->seq_len_ * lstm_param_->batch_ * lstm_param_->hidden_si…
8101 +  float *cell_gate = gate + lstm_param_->seq_len_ * lstm_param_->batch_ * lstm_param_->hidden_size…
8102 +  float *output_gate = gate + lstm_param_->seq_len_ * lstm_param_->batch_ * lstm_param_->hidden_si…
8189 + *    weight_bias: forth input, weight_ih + weight_hh + bias, the gate order is IFGO
8194 + *    weight_bias: forth input, weight_ih + weight_hh + proj + bias, the gate order is IFGO
8386 +  float *gate = buffer[kInputGateIndex];
8387 +  float *input_gate = gate;
8388 +  float *forget_gate = gate + lstm_param_->seq_len_ * lstm_param_->batch_ * lstm_param_->hidden_si…
8389 +  float *cell_gate = gate + lstm_param_->seq_len_ * lstm_param_->batch_ * lstm_param_->hidden_size…
8390 +  float *output_gate = gate + lstm_param_->seq_len_ * lstm_param_->batch_ * lstm_param_->hidden_si…