Searched refs:_compute_static_batch_dim (Results 1 – 1 of 1) sorted by relevance
138 new_batch_dim = self._compute_static_batch_dim()154 def _compute_static_batch_dim(self): member in _RebatchDataset