Searched refs:TPUDatasetInfeedManager (Results 1 – 1 of 1) sorted by relevance
706 class TPUDatasetInfeedManager(TPUInfeedManager): class1515 infeed_manager = TPUDatasetInfeedManager(1536 infeed_manager = TPUDatasetInfeedManager(dataset, self._tpu_assignment,1598 infeed_manager = TPUDatasetInfeedManager(dataset, self._tpu_assignment,