diff options
author | Russell Power <power@google.com> | 2018-08-07 14:30:02 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-08-07 14:36:52 -0700 |
commit | cf233f7281fbb841f5ea2df548e6e857308f222b (patch) | |
tree | 5766a9194a644d14ea414a5c5bddc6bf742f3ae7 | |
parent | 02df0f46d562a0c48b6f24803eba6330d13d7213 (diff) |
Enable infeed enqueue, outfeed dequeue operations.
PiperOrigin-RevId: 207780309
-rw-r--r-- | tensorflow/contrib/tpu/__init__.py | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/tensorflow/contrib/tpu/__init__.py b/tensorflow/contrib/tpu/__init__.py index d0a37eb0ed..537d94b797 100644 --- a/tensorflow/contrib/tpu/__init__.py +++ b/tensorflow/contrib/tpu/__init__.py @@ -18,6 +18,10 @@ @@cross_replica_sum @@infeed_dequeue @@infeed_dequeue_tuple +@@infeed_enqueue +@@infeed_enqueue_tuple +@@outfeed_dequeue +@@outfeed_dequeue_tuple @@outfeed_enqueue @@outfeed_enqueue_tuple |