diff options
-rw-r--r-- | tensorflow/core/common_runtime/renamed_device.h | 9 |
1 files changed, 9 insertions, 0 deletions
diff --git a/tensorflow/core/common_runtime/renamed_device.h b/tensorflow/core/common_runtime/renamed_device.h index 9d59264899..c00789a556 100644 --- a/tensorflow/core/common_runtime/renamed_device.h +++ b/tensorflow/core/common_runtime/renamed_device.h @@ -58,6 +58,15 @@ class RenamedDevice : public Device { return underlying_->GetAllocator(attr); } + Allocator* GetScopedAllocator(AllocatorAttributes attr, + int64 step_id) override { + return underlying_->GetScopedAllocator(attr, step_id); + } + + ScopedAllocatorMgr* GetScopedAllocatorMgr() const override { + return underlying_->GetScopedAllocatorMgr(); + } + const Eigen::ThreadPoolDevice* eigen_cpu_device() override { return underlying_->eigen_cpu_device(); } |