From 5d51afdfec8c6a96d48457d4678e2835100577a6 Mon Sep 17 00:00:00 2001 From: Bairen Yi Date: Tue, 18 Sep 2018 13:17:07 +0800 Subject: Support scoped_allocator_ops for renamed device. This fixes #22274. Signed-off-by: Bairen Yi --- tensorflow/core/common_runtime/renamed_device.h | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/tensorflow/core/common_runtime/renamed_device.h b/tensorflow/core/common_runtime/renamed_device.h index 103eee03b3..caf1300d85 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(); } -- cgit v1.2.3