From cb4c2f8d16d70ea9de09d29d82934dbb25359700 Mon Sep 17 00:00:00 2001 From: Yifei Feng Date: Wed, 21 Jun 2017 16:49:43 -0700 Subject: [PATCH] Make TransferBufferToInFeed not virual so it compiles. --- tensorflow/compiler/plugin/executor/transfer_manager.cc | 5 +++++ tensorflow/compiler/plugin/executor/transfer_manager.h | 3 +++ 2 files changed, 8 insertions(+) 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;