Searched refs:TPUNumpyInfeedManager (Results 1 – 1 of 1) sorted by relevance
613 class TPUNumpyInfeedManager(TPUInfeedManager): class1157 return TPUNumpyInfeedManager(self.model._tpu_assignment)