提交 a3d1f869 编写于 作者: Y Yibing Liu

Add unit test for linear projection

上级 7a5b8ffa
......@@ -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()
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册