From 6c8cb9bd76299be00d787be2ba02daed0b3921a6 Mon Sep 17 00:00:00 2001
From: Kevin Tse <ktse@fb.com>
Date: Tue, 31 Aug 2021 13:55:59 -0700
Subject: [PATCH] [DataPipe] export fork, mux, demux for public usage (#64279)

Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/64279

cc VitalyFedyunin ejguan

Test Plan: Imported from OSS

Reviewed By: ejguan

Differential Revision: D30671971

Pulled By: NivekT

fbshipit-source-id: 056ac12ef7183b254d1eec341145594639e47ef6
---
 torch/utils/data/datapipes/iter/__init__.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/torch/utils/data/datapipes/iter/__init__.py b/torch/utils/data/datapipes/iter/__init__.py
index d4baef7..26d715d 100644
--- a/torch/utils/data/datapipes/iter/__init__.py
+++ b/torch/utils/data/datapipes/iter/__init__.py
@@ -54,14 +54,17 @@ __all__ = ['Batcher',
            'BucketBatcher',
            'Collator',
            'Concater',
+           'Demultiplexer',
            'FileLister',
            'FileLoader',
            'Filter',
+           'Forker',
            'Grouper',
            'HttpReader',
            'IterableWrapper',
            'LineReader',
            'Mapper',
+           'Multiplexer',
            'RoutedDecoder',
            'Sampler',
            'Shuffler',
-- 
2.7.4