Searched refs:_PerDeviceGenerator (Results 1 – 1 of 1) sorted by relevance
34 class _PerDeviceGenerator(dataset_ops.DatasetV2): class136 super(_PerDeviceGenerator, self).__init__(variant_tensor)254 ds = _PerDeviceGenerator(