Searched refs:_get_output_shape (Results 1 – 3 of 3) sorted by relevance
103 output_shape = _get_output_shape(113 output_shape = _get_output_shape(123 output_shape = _get_output_shape(132 output_shape = _get_output_shape(138 def _get_output_shape(model_fn): function
122 def _get_output_shape(output_rank, known_last_dims): function352 output_shape=_get_output_shape(output_rank - 1,361 output_shape=_get_output_shape(output_rank - 1,370 output_shape=_get_output_shape(output_rank - 1,391 output_shape=_get_output_shape(output_rank - 1, output_shape),
493 def _get_output_shape(flat_output_size): function509 _get_output_shape, self.cell.output_size))513 output_shape = _get_output_shape(state_size[0])