Searched refs:_get_output_shapes_from_input_shapes (Results 1 – 1 of 1) sorted by relevance
452 per_replica_output_shapes = self._get_output_shapes_from_input_shapes(476 def _get_output_shapes_from_input_shapes( member in TPUEmbedding1247 self._get_output_shapes_from_input_shapes(input_shapes))