Lines Matching refs:flatten
41 for i, t in enumerate(nest.flatten(initial_state))
50 [t.shape for t in nest.flatten(self._initial_state)])
53 [t.dtype for t in nest.flatten(self._initial_state)])
65 flat_state_shapes = nest.flatten(self._state_shapes)
66 flat_state_types = nest.flatten(self._state_types)
71 @function.Defun(*(flat_state_types + nest.flatten(
81 flat_state_shapes + nest.flatten(dense_shapes)):
96 ops.convert_to_tensor(t) for t in nest.flatten(new_state)
99 ops.convert_to_tensor(t) for t in nest.flatten(output_value)
152 nest.flatten(self._initial_state),
155 output_types=nest.flatten(
157 output_shapes=nest.flatten(
