Searched refs:freed_by_func (Results 1 – 7 of 7) sorted by relevance
43 std::function<uint64()>* freed_by_func) in AllocationAttributes()46 freed_by_func(freed_by_func) {} in AllocationAttributes()64 std::function<uint64()>* freed_by_func = nullptr; // Not owned. member
714 /*allocation_will_be_logged=*/true, allocation_attr.freed_by_func)); in allocate_tensor()
88 std::function<uint64()> freed_by_func = [dst_device, in SameWorkerRecvDone() local96 aa.freed_by_func = &freed_by_func; in SameWorkerRecvDone()
132 std::function<uint64()> freed_by_func = [this, &safe_alloc_frontier]() { in ContinueAfterInputCopy() local138 aa.freed_by_func = &freed_by_func; in ContinueAfterInputCopy()
231 if (allocation_attr.freed_by_func != nullptr) { in AllocateRawInternalWithRetry()232 freed_by_count = (*allocation_attr.freed_by_func)(); in AllocateRawInternalWithRetry()243 if (allocation_attr.freed_by_func != nullptr) { in AllocateRawInternalWithRetry()244 freed_by_count = (*allocation_attr.freed_by_func)(); in AllocateRawInternalWithRetry()261 if (allocation_attr.freed_by_func != nullptr) { in AllocateRaw()262 freed_by_count = (*allocation_attr.freed_by_func)(); in AllocateRaw()
271 std::function<uint64()> freed_by_func = [dst_device, &safe_alloc_frontier]() { in SameWorkerRecvDone() local276 allocation_attr.freed_by_func = &freed_by_func; in SameWorkerRecvDone()
650 std::function<uint64()> freed_by_func = [this, &safe_alloc_frontier]() { in MaybeCopyTensorToGPU() local655 allocation_attr.freed_by_func = &freed_by_func; in MaybeCopyTensorToGPU()