Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/common_runtime/
Drenamed_device.h62 Allocator* GetScopedAllocator(AllocatorAttributes attr, in GetScopedAllocator() function
Dthreadpool_device.cc110 Allocator* ThreadPoolDevice::GetScopedAllocator(AllocatorAttributes attr, in GetScopedAllocator() function in tensorflow::ThreadPoolDevice
/external/tensorflow/tensorflow/core/framework/
Ddevice_base.h199 virtual Allocator* GetScopedAllocator(AllocatorAttributes attr, in GetScopedAllocator() function
Dop_kernel_test.cc404 Allocator* GetScopedAllocator(AllocatorAttributes attrs, in GetScopedAllocator() function in tensorflow::__anon1077b3a10111::ScopedAllocatorDevice
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dlight_outside_compilation.cc340 Allocator* GetScopedAllocator(AllocatorAttributes attrs, in GetScopedAllocator() function in tensorflow::__anonfdc132ad0311::TfCallbackDevice
/external/tensorflow/tensorflow/core/runtime_fallback/kernel/
Dkernel_fallback_execute_compat.cc781 Allocator* GetScopedAllocator(AllocatorAttributes attr, in GetScopedAllocator() function in tensorflow::tfd::__anone1369dfd0711::DeviceWithCustomAllocator
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_device.cc1193 Allocator* BaseGPUDevice::GetScopedAllocator(AllocatorAttributes attr, in GetScopedAllocator() function in tensorflow::BaseGPUDevice