diff --git a/tensorflow/compiler/plugin/executor/transfer_manager.cc b/tensorflow/compiler/plugin/executor/transfer_manager.cc index 6a471401fae..51c5deeea5d 100644 --- a/tensorflow/compiler/plugin/executor/transfer_manager.cc +++ b/tensorflow/compiler/plugin/executor/transfer_manager.cc @@ -147,6 +147,11 @@ Status ExecutorTransferManager::TransferLiteralToInfeed( return Status::OK(); } +Status ExecutorTransferManager::TransferBufferToInfeed( + se::StreamExecutor* executor, int64 size, const void* source) { + return Unimplemented("Transfer to Infeed"); +} + Status ExecutorTransferManager::TransferLiteralFromOutfeed( perftools::gputools::StreamExecutor* executor, const Shape& literal_shape, Literal* literal) { diff --git a/tensorflow/compiler/plugin/executor/transfer_manager.h b/tensorflow/compiler/plugin/executor/transfer_manager.h index 22142cd778a..7a42e5a2d75 100644 --- a/tensorflow/compiler/plugin/executor/transfer_manager.h +++ b/tensorflow/compiler/plugin/executor/transfer_manager.h @@ -55,6 +55,9 @@ class ExecutorTransferManager : public TransferManager { Status TransferLiteralToInfeed(se::StreamExecutor* executor, const Literal& literal) override; + Status TransferBufferToInfeed(se::StreamExecutor* executor, + int64 size, const void* source) override; + Status TransferLiteralFromOutfeed(se::StreamExecutor* executor, const Shape& literal_shape, Literal* literal) override;