Searched refs:_split_tensors (Results 1 – 1 of 1) sorted by relevance
648 def _split_tensors(self, inputs): member in TPUNumpyInfeedManager676 sharded_inputs = self._split_tensors(inputs)