Home
last modified time | relevance | path

Searched defs:backward_output (Results 1 – 5 of 5) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/
Dgru_fp16.c138 float16_t *backward_output = output + gru_param->batch_ * gru_param->hidden_size_; in GruFp16() local
Dlstm_fp16.c291 float16_t *backward_output = output + lstm_param->batch_ * lstm_param->hidden_size_; in LstmFp16() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/
Dgru_fp32.c143 float *backward_output = output + gru_param->batch_ * gru_param->hidden_size_; in Gru() local
Dlstm_fp32.c276 float *backward_output = output + lstm_param->batch_ * lstm_param->hidden_size_; in Lstm() local
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/
Dlstm_fp32.cc408 float *backward_output = output + lstm_param_->batch_ * lstm_param_->hidden_size_; in InnerExecute() local