(self, cell_class, dtype, batch_size, in_size,
out_size)
| 338 | self.assertEqual(initial_state.dtype, inputs.dtype) |
| 339 | |
| 340 | def _assert_cell_builds(self, cell_class, dtype, batch_size, in_size, |
| 341 | out_size): |
| 342 | cell = cell_class(out_size, dtype=dtype) |
| 343 | in_shape = tensor_shape.TensorShape((batch_size, in_size)) |
| 344 | cell.build(in_shape) |
| 345 | state_output = cell.get_initial_state( |
| 346 | inputs=None, batch_size=batch_size, dtype=dtype) |
| 347 | cell_output, _ = cell(array_ops.zeros(in_shape, dtype), state_output) |
| 348 | self.assertAllEqual([batch_size, out_size], cell_output.shape.as_list()) |
| 349 | |
| 350 | @test_util.run_in_graph_and_eager_modes |
| 351 | def testCellsBuild(self): |
no test coverage detected