Make TransferBufferToInFeed not virual so it compiles.

This commit is contained in:
Yifei Feng 2017-06-21 16:49:43 -07:00
parent e89f04d801
commit cb4c2f8d16
2 changed files with 8 additions and 0 deletions
tensorflow/compiler/plugin/executor

View File

@ -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) {

View File

@ -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;