Searched refs:_pad_or_truncate_with_sequence_length (Results 1 – 1 of 1) sorted by relevance
216 def _pad_or_truncate_with_sequence_length(self, embeddings: ops.Tensor, member in TPUEmbeddingV0344 embeddings = self._pad_or_truncate_with_sequence_length(