diff options
author | TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-20 16:44:24 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-20 16:44:24 -0700 |
commit | 863b3bd6bb2214065f95fa20f551a8fc8568e55d (patch) | |
tree | 86b86c1b74eb03f2e422fc6825a78ab4640d4aa1 | |
parent | 23a88ec5e913ba7086a9aef57875447ccf96e4b5 (diff) | |
parent | 5d51afdfec8c6a96d48457d4678e2835100577a6 (diff) |
Merge pull request #22337 from byronyi:scoped_allocator_ops_fix
PiperOrigin-RevId: 213906379
-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(); } |