Add experimental parallel iterators API (#6644)

This commit is contained in:
Eric Liang
2020-01-02 13:45:26 -08:00
committed by GitHub
parent 3dddbef6d9
commit 895f2727fb
6 changed files with 1070 additions and 3 deletions
+12 -3
View File
@@ -8,6 +8,7 @@ from .named_actors import get_actor, register_actor
from .api import get, wait
from .actor_pool import ActorPool
from .dynamic_resources import set_resource
from . import iter
def TensorFlowVariables(*args, **kwargs):
@@ -17,7 +18,15 @@ def TensorFlowVariables(*args, **kwargs):
__all__ = [
"TensorFlowVariables", "get_actor", "register_actor", "get", "wait",
"set_flushing_policy", "GcsFlushPolicy", "SimpleGcsFlushPolicy",
"set_resource", "ActorPool"
"TensorFlowVariables",
"get_actor",
"register_actor",
"get",
"wait",
"set_flushing_policy",
"GcsFlushPolicy",
"SimpleGcsFlushPolicy",
"set_resource",
"ActorPool",
"iter",
]