Searched refs:_RemoteDataset (Results 1 – 2 of 2) sorted by relevance
297 class _RemoteDataset(dataset_ops.DatasetSource): class304 super(_RemoteDataset, self).__init__(variant_tensor)348 ds = _RemoteDataset(graph_def, device, dataset.element_spec)
328 ds1 = distribute._RemoteDataset(graph_def, "/device:CPU:0",330 ds2 = distribute._RemoteDataset(graph_def, "/device:CPU:0",