diff --git a/python/paddle/v2/fluid/tests/test_lstmp_op.py b/python/paddle/v2/fluid/tests/test_lstmp_op.py index a0f6955d7728770158ae25940844f76911d5594f..8835cae504cb2fc74642167f19b15e6d20cd4c2f 100644 --- a/python/paddle/v2/fluid/tests/test_lstmp_op.py +++ b/python/paddle/v2/fluid/tests/test_lstmp_op.py @@ -192,7 +192,8 @@ class TestLstmOp(OpTest): 'is_reverse': self.is_reverse, 'gate_activation': self.act_gate, 'cell_activation': self.act_cell, - 'candidate_activation': self.act_cand + 'candidate_activation': self.act_cand, + 'share_cell_act': self.share_cell_act } def test_check_output(self): @@ -340,9 +341,25 @@ class TestLstmOpNotUsePeepholes(TestLstmOp): self.share_cell_act = True self.has_initial_state = False - self.is_reverse = True + self.is_reverse = False self.use_peepholes = False +class TestLstmOpNotShareCellAct(TestLstmOp): + def set_argument(self): + self.lod = [[0, 2, 5, 7]] + self.D = 16 + self.P = 10 + + self.act_gate = 'sigmoid' + self.act_cell = 'tanh' + self.act_cand = 'tanh' + + self.share_cell_act = False + self.has_initial_state = False + self.is_reverse = False + self.use_peepholes = True + + if __name__ == '__main__': unittest.main()