Home
last modified time | relevance | path

Searched defs:GetScopedAllocatorMgr (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dthreadpool_device.h36 ScopedAllocatorMgr* GetScopedAllocatorMgr() const override { in GetScopedAllocatorMgr() function
Drenamed_device.h67 ScopedAllocatorMgr* GetScopedAllocatorMgr() const override { in GetScopedAllocatorMgr() function
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_device.h127 ScopedAllocatorMgr* GetScopedAllocatorMgr() const override { in GetScopedAllocatorMgr() function
/external/tensorflow/tensorflow/core/framework/
Ddevice_base.h205 virtual ScopedAllocatorMgr* GetScopedAllocatorMgr() const { return nullptr; } in GetScopedAllocatorMgr() function
/external/tensorflow/tensorflow/core/runtime_fallback/kernel/
Dkernel_fallback_execute_compat.cc786 ScopedAllocatorMgr* GetScopedAllocatorMgr() const override { in GetScopedAllocatorMgr() function in tensorflow::tfd::__anon0fecb2ce0711::DeviceWithCustomAllocator