Searched refs:AddAllocationOrIncrementRefCount (Results 1 – 2 of 2) sorted by relevance
87 AddAllocationOrIncrementRefCount(shaped_buffer.buffer(index), in RegisterInternal()218 void AllocationTracker::AddAllocationOrIncrementRefCount( in AddAllocationOrIncrementRefCount() function in xla::AllocationTracker
103 void AddAllocationOrIncrementRefCount(se::DeviceMemoryBase device_memory,