Searched refs:output_gate (Results 1 – 6 of 6) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/ |
| D | lstm_fp32.c | 137 void UpdataOutput(const float *cell_state, const float *output_gate, float *hidden_state, float *st… in UpdataOutput() argument
148 ElementMul(hidden_state, output_gate, hidden_state, batch * hidden_size); in UpdataOutput()
177 …tepUnit(float *output, float *input_gate, float *forget_gate, float *cell_gate, float *output_gate, in LstmStepUnit() argument
201 ElementAdd(output_gate, state_gate + lstm_param->batch_ * lstm_param->hidden_size_, output_gate, in LstmStepUnit()
217 Sigmoid(output_gate, lstm_param->batch_ * lstm_param->hidden_size_, output_gate); in LstmStepUnit()
219 …UpdataOutput(cell_state, output_gate, hidden_state, hidden_buffer, lstm_param->batch_, lstm_param-… in LstmStepUnit()
248 float *output_gate = gate + lstm_param->seq_len_ * lstm_param->batch_ * lstm_param->hidden_size_; in LstmUnidirectional() local
254 float *output_gate_t = output_gate + lstm_param->batch_ * lstm_param->hidden_size_ * real_t; in LstmUnidirectional()
|
| D | lstm_fp32.h | 37 …tepUnit(float *output, float *input_gate, float *forget_gate, float *cell_gate, float *output_gate,
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/ |
| D | lstm_fp16.c | 153 void UpdataOutputFp16(const float16_t *cell_state, float16_t *output_gate, float16_t *hidden_state, in UpdataOutputFp16() argument
164 ElementMulFp16(hidden_state, output_gate, hidden_state, batch * hidden_size); in UpdataOutputFp16()
192 … float16_t *output_gate, const float16_t *state_weight, const float16_t *state_bias, in LstmStepUnitFp16() argument
214 …ElementAddFp16(output_gate, state_gate + lstm_param->batch_ * lstm_param->hidden_size_, output_gat… in LstmStepUnitFp16()
230 SigmoidFp16(output_gate, output_gate, lstm_param->batch_ * lstm_param->hidden_size_); in LstmStepUnitFp16()
232 …UpdataOutputFp16(cell_state, output_gate, hidden_state, hidden_buffer, lstm_param->batch_, lstm_pa… in LstmStepUnitFp16()
262 …float16_t *output_gate = gate + lstm_param->seq_len_ * lstm_param->batch_ * lstm_param->hidden_siz… in LstmUnidirectionalFp16() local
268 float16_t *output_gate_t = output_gate + lstm_param->batch_ * lstm_param->hidden_size_ * real_t; in LstmUnidirectionalFp16()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/ |
| D | lstm_fp32.cc | 366 …float *output_gate = gate + lstm_param_->seq_len_ * lstm_param_->batch_ * lstm_param_->hidden_size… in LstmUnidirectional() local
372 float *output_gate_t = output_gate + lstm_param_->batch_ * lstm_param_->hidden_size_ * real_t; in LstmUnidirectional()
|
| /third_party/mindspore/mindspore/lite/tools/optimizer/fusion/ |
| D | tf_lstm_cell_fusion.cc | 89 VectorRef output_gate = VectorRef({is_var_sigmoid3, get_items[3]}); in GenerateBodyGraphHiddenPattern() local
105 VectorRef new_hidden = VectorRef({is_var_mul, output_gate, to_new_hidden}); in GenerateBodyGraphHiddenPattern()
|
| D | tflite_lstm_cell_fusion.cc | 246 VectorRef output_gate = VectorRef({is_var_sigmoid, bias_output}); in GetBodyGraphPattern() local
252 VectorRef output = VectorRef({is_var_mul3, output_gate, cell_to_output}); in GetBodyGraphPattern()
|