Searched refs:GetAllocatorLocked (Results 1 – 2 of 2) sorted by relevance
254 return GetAllocatorLocked(attr); in GetAllocator()257 Allocator* XlaDevice::GetAllocatorLocked(AllocatorAttributes attr) { in GetAllocatorLocked() function in tensorflow::XlaDevice332 GetAllocatorLocked({}); in GetDeviceContextLocked()489 Allocator* allocator = GetAllocatorLocked(alloc_attrs); in MakeTensorFromProto()
193 Allocator* GetAllocatorLocked(AllocatorAttributes attr)