xxq250
/
mindspore
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

!2106 fix BasicLSTMCell reg and bp error 

Merge pull request !2106 from zhaozhenlong/fix-issue-lstm-reg-error
This commit is contained in:
mindspore-ci-bot 2020-06-15 16:12:10 +08:00 committed by Gitee
parent 5845c4ad0d c3445bfb2d
commit a0b346690f
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/ops/_grad/grad_nn_ops.py
View File

@ -714,7 +714,7 @@ def get_bprop_basic_lstm_cell(self):
def bprop(x, h, c, w, b, out, dout):
_, _, it, jt, ft, ot, tanhct = out
dct, dht, _, _, _, _, _ = dout
dgate, dct_1 = basic_lstm_cell_cstate_grad(c, dht, dct, it, jt, ft, ot, tanhct)
dgate, dct_1 = basic_lstm_cell_cstate_grad(c, dht, dct, it, ft, jt, ot, tanhct)
dxt, dht = basic_lstm_cell_input_grad(dgate, w)
dw, db = basic_lstm_cell_weight_grad(F.depend(x, dxt), h, dgate)
return dxt, dht, dct_1, dw, db

4
mindspore/ops/_op_impl/tbe/basic_lstm_cell_c_state_grad.py
View File

@ -37,10 +37,10 @@ basic_lstm_cell_c_state_grad_op_info = TBERegOp("BasicLSTMCellCStateGrad") \
.output(1, "dct_1", False, "required", "all") \
.dtype_format(DataType.F32_FracNZ, DataType.F32_FracNZ, DataType.F32_FracNZ, DataType.F32_FracNZ,
DataType.F32_FracNZ, DataType.F32_FracNZ, DataType.F32_FracNZ, DataType.F32_FracNZ,
DataType.F16_FracNZ, DataType.F16_FracNZ) \
DataType.F16_FracNZ, DataType.F32_FracNZ) \
.dtype_format(DataType.F16_FracNZ, DataType.F16_FracNZ, DataType.F16_FracNZ, DataType.F16_FracNZ,
DataType.F16_FracNZ, DataType.F16_FracNZ, DataType.F16_FracNZ, DataType.F16_FracNZ,
DataType.F32_FracNZ, DataType.F16_FracNZ) \
DataType.F16_FracNZ, DataType.F16_FracNZ) \
.get_op_info()