Searched refs:GetAllocatorLocked (Results 1 – 2 of 2) sorted by relevance
242 return GetAllocatorLocked(attr); in GetAllocator()245 Allocator* XlaDevice::GetAllocatorLocked(AllocatorAttributes attr) { in GetAllocatorLocked() function in tensorflow::XlaDevice316 GetAllocatorLocked({}); in GetDeviceContextLocked()458 Allocator* allocator = GetAllocatorLocked(alloc_attrs); in MakeTensorFromProto()
181 Allocator* GetAllocatorLocked(AllocatorAttributes attr)