(cell, w, i)
| 1514 | def _BuildForward(self, weights, inp, mode="cell"): |
| 1515 | |
| 1516 | def Loop(cell, w, i): |
| 1517 | x = array_ops.unstack(i, self.NUM_UNROLL) |
| 1518 | m = array_ops.zeros_like(x[0]) |
| 1519 | c = array_ops.zeros_like(x[0]) |
| 1520 | for i in range(self.NUM_UNROLL): |
| 1521 | m, c = cell(x[i], m, c, w) |
| 1522 | return m |
| 1523 | |
| 1524 | cell = UnrollLSTMTest.LSTMCell |
| 1525 | if mode == "complete": |
no test coverage detected