1 files changed, 4 insertions, 2 deletions
diff --git a/tensorflow/core/common_runtime/sycl/sycl_allocator.h b/tensorflow/core/common_runtime/sycl/sycl_allocator.h
index 887c727f6e..8558b6c873 100644
--- a/tensorflow/core/common_runtime/sycl/sycl_allocator.h
+++ b/tensorflow/core/common_runtime/sycl/sycl_allocator.h
@@ -29,14 +29,16 @@ namespace tensorflow {
 
 class SYCLAllocator : public Allocator {
 public:
-  SYCLAllocator(Eigen::SyclDevice* device) : device_(device) {}
+  SYCLAllocator(Eigen::QueueInterface* device) : device_(device) {}
   virtual ~SYCLAllocator() override;
   string Name() override;
   void *AllocateRaw(size_t alignment, size_t num_bytes) override;
   void DeallocateRaw(void *ptr) override;
 
+  void EnterLameDuckMode();
+  virtual bool ShouldAllocateEmptyTensors() override final { return true; }
 private:
-  Eigen::SyclDevice *device_;  // not owned
+  Eigen::QueueInterface *device_;  // not owned
   TF_DISALLOW_COPY_AND_ASSIGN(SYCLAllocator);
 };