Home
last modified time | relevance | path

Searched defs:GetScopedAllocator (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dthreadpool_device.cc80 Allocator* ThreadPoolDevice::GetScopedAllocator(AllocatorAttributes attr, in GetScopedAllocator() function in tensorflow::ThreadPoolDevice
Drenamed_device.h62 Allocator* GetScopedAllocator(AllocatorAttributes attr, in GetScopedAllocator() function
/external/tensorflow/tensorflow/core/framework/
Ddevice_base.h196 virtual Allocator* GetScopedAllocator(AllocatorAttributes attr, in GetScopedAllocator() function
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_device.cc964 Allocator* BaseGPUDevice::GetScopedAllocator(AllocatorAttributes attr, in GetScopedAllocator() function in tensorflow::BaseGPUDevice