Use the cached allocator pointer rather than looking it up again when creating XlaScopedShapedBuffers and ScopedDeviceMemories.
Use the cached allocator pointer rather than looking it up again when creating XlaScopedShapedBuffers and ScopedDeviceMemories.
