diff options
| author |  Jacques Pienaar <jpienaar@google.com> | 2017-02-27 15:05:10 -0800 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2017-02-27 15:42:08 -0800 |
| commit | efc8f98d45df835bac2373e19f1da57e3a1ea2d0 (patch) | |
| tree | 28dabbd42fb3955e6569c0ae1d6954b14433d338 /tensorflow/compiler/xla/service/generic_transfer_manager.h | |
| parent | 28554fbc756454cd2a1f6f6bda2b2cc86c68bcff (diff) | |
[XLA] Add basic outfeed support.
Change: 148699787
Diffstat (limited to 'tensorflow/compiler/xla/service/generic_transfer_manager.h')
| -rw-r--r-- | tensorflow/compiler/xla/service/generic_transfer_manager.h | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/tensorflow/compiler/xla/service/generic_transfer_manager.h b/tensorflow/compiler/xla/service/generic_transfer_manager.h index 06819d65c7..2fbdb94f06 100644 --- a/tensorflow/compiler/xla/service/generic_transfer_manager.h +++ b/tensorflow/compiler/xla/service/generic_transfer_manager.h @@ -55,6 +55,10 @@ class GenericTransferManager : public TransferManager { Status TransferLiteralToInfeed(perftools::gputools::StreamExecutor* executor, const Literal& literal) override; + Status TransferLiteralFromOutfeed( + perftools::gputools::StreamExecutor* executor, const Shape& literal_shape, + Literal* literal) override; + Status ResetDevices( tensorflow::gtl::ArraySlice<perftools::gputools::StreamExecutor*> executors) override; |