(self, op, w, b, x, cs_prev, h_prev)
| 67 | self.assertAllEqual(b_grad, b_ifco_grad) |
| 68 | |
| 69 | def _lstm_block(self, op, w, b, x, cs_prev, h_prev): |
| 70 | w_peephole = array_ops.zeros(cs_prev.shape[1:], dtype=w.dtype) |
| 71 | _, all_cs, _, _, _, _, all_h = op( |
| 72 | seq_len_max=math_ops.cast(array_ops.shape(x)[0], dtypes.int64), |
| 73 | x=x, |
| 74 | cs_prev=cs_prev, |
| 75 | h_prev=h_prev, |
| 76 | w=w, |
| 77 | wci=w_peephole, |
| 78 | wcf=w_peephole, |
| 79 | wco=w_peephole, |
| 80 | b=b, |
| 81 | use_peephole=False) |
| 82 | return all_cs, all_h |
| 83 | |
| 84 | |
| 85 | def deterministic_random_uniform(shape): |
no test coverage detected