diff options
author | Jacques Pienaar <jpienaar@google.com> | 2017-06-15 14:16:59 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-06-15 14:20:48 -0700 |
commit | b52debb4e63cce1e0733d6d34975d4efb9934680 (patch) | |
tree | 6cfc65a41e8c98b938272f2df6436e1b9a48f8dd /tensorflow/compiler/xla/service/generic_transfer_manager.h | |
parent | a203bbb98c7c72d7a68064098ec051bba20219cd (diff) |
[XLA] Add transfer buffer to infeed.
Mirroring the transfer buffer to device interface, add a transfer buffer to
infeed interface.
PiperOrigin-RevId: 159152897
Diffstat (limited to 'tensorflow/compiler/xla/service/generic_transfer_manager.h')
-rw-r--r-- | tensorflow/compiler/xla/service/generic_transfer_manager.h | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/compiler/xla/service/generic_transfer_manager.h b/tensorflow/compiler/xla/service/generic_transfer_manager.h index 2fbdb94f06..48c061d28e 100644 --- a/tensorflow/compiler/xla/service/generic_transfer_manager.h +++ b/tensorflow/compiler/xla/service/generic_transfer_manager.h @@ -54,6 +54,8 @@ class GenericTransferManager : public TransferManager { Status TransferLiteralToInfeed(perftools::gputools::StreamExecutor* executor, const Literal& literal) override; + Status TransferBufferToInfeed(perftools::gputools::StreamExecutor* executor, + int64 size, const void* source) override; Status TransferLiteralFromOutfeed( perftools::gputools::StreamExecutor* executor, const Shape& literal_shape, |