diff --git a/tensorflow/compiler/xla/service/cpu/cpu_transfer_manager.h b/tensorflow/compiler/xla/service/cpu/cpu_transfer_manager.h index 361d4b9c842..d933380442f 100644 --- a/tensorflow/compiler/xla/service/cpu/cpu_transfer_manager.h +++ b/tensorflow/compiler/xla/service/cpu/cpu_transfer_manager.h @@ -44,6 +44,12 @@ class CpuTransferManager : public GenericTransferManager { const Shape& literal_shape, MutableBorrowingLiteral literal) override; + bool CanShapedBufferBeAccessedNow( + se::StreamExecutor* executor, + const ShapedBuffer& device_buffer) const override { + return true; + } + private: Status TransferBufferToInfeed(se::StreamExecutor* executor, int64 size, const void* source);