Searched refs:_get_batch_size (Results 1 – 1 of 1) sorted by relevance
1239 batch_size = self._get_batch_size(features, in_tpu_context)1336 def _get_batch_size(self, tensors, in_tpu_context: bool): member in TPUEmbedding