Searched refs:GetScopedAllocator (Results 1 – 8 of 8) sorted by relevance
62 Allocator* GetScopedAllocator(AllocatorAttributes attr, in GetScopedAllocator() function64 return underlying_->GetScopedAllocator(attr, step_id); in GetScopedAllocator()
33 Allocator* GetScopedAllocator(AllocatorAttributes attr,
80 Allocator* ThreadPoolDevice::GetScopedAllocator(AllocatorAttributes attr, in GetScopedAllocator() function in tensorflow::ThreadPoolDevice
196 virtual Allocator* GetScopedAllocator(AllocatorAttributes attr, in GetScopedAllocator() function
320 allocator = params_->device->GetScopedAllocator(attr, step_id()); in get_allocator()
112 Allocator* GetScopedAllocator(AllocatorAttributes attr,
964 Allocator* BaseGPUDevice::GetScopedAllocator(AllocatorAttributes attr, in GetScopedAllocator() function in tensorflow::BaseGPUDevice
59 allocator = device_->GetScopedAllocator(attr, context_->step_id()); in MakeOp()