Searched refs:NumpyInfeedInstance (Results 1 – 1 of 1) sorted by relevance
616 class NumpyInfeedInstance(TPUInfeedInstance): class in TPUNumpyInfeedManager677 return self.NumpyInfeedInstance(sharded_inputs)