experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Make TransferBufferToInFeed not virual so it compiles.

Browse Source
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tensorflow/compiler/plugin/executor/transfer_manager.cc
View File

@ -147,6 +147,11 @@ Status ExecutorTransferManager::TransferLiteralToInfeed(
return Status::OK(); return Status::OK();
} }
Status ExecutorTransferManager::TransferBufferToInfeed(
se::StreamExecutor* executor, int64 size, const void* source) {
return Unimplemented("Transfer to Infeed");
}
Status ExecutorTransferManager::TransferLiteralFromOutfeed( Status ExecutorTransferManager::TransferLiteralFromOutfeed(
perftools::gputools::StreamExecutor* executor, const Shape& literal_shape, perftools::gputools::StreamExecutor* executor, const Shape& literal_shape,
Literal* literal) { Literal* literal) {

3
tensorflow/compiler/plugin/executor/transfer_manager.h
View File

@ -55,6 +55,9 @@ class ExecutorTransferManager : public TransferManager {
Status TransferLiteralToInfeed(se::StreamExecutor* executor, Status TransferLiteralToInfeed(se::StreamExecutor* executor,
const Literal& literal) override; const Literal& literal) override;
Status TransferBufferToInfeed(se::StreamExecutor* executor,
int64 size, const void* source) override;
Status TransferLiteralFromOutfeed(se::StreamExecutor* executor, Status TransferLiteralFromOutfeed(se::StreamExecutor* executor,
const Shape& literal_shape, const Shape& literal_shape,
Literal* literal) override; Literal* literal) override;