Searched refs:freed_by_func (Results 1 – 7 of 7) sorted by relevance
41 std::function<uint64()>* freed_by_func) in AllocationAttributes()44 freed_by_func(freed_by_func) {} in AllocationAttributes()60 std::function<uint64()>* freed_by_func = nullptr; // Not owned. member
756 allocation_attr.freed_by_func)); in allocate_tensor()
87 std::function<uint64()> freed_by_func = [dst_device, in SameWorkerRecvDone() local95 aa.freed_by_func = &freed_by_func; in SameWorkerRecvDone()
133 std::function<uint64()> freed_by_func = [this, &safe_alloc_frontier]() { in ContinueAfterInputCopy() local139 aa.freed_by_func = &freed_by_func; in ContinueAfterInputCopy()
205 if (allocation_attr.freed_by_func != nullptr) { in AllocateRawInternalWithRetry()206 freed_by_count = (*allocation_attr.freed_by_func)(); in AllocateRawInternalWithRetry()217 if (allocation_attr.freed_by_func != nullptr) { in AllocateRawInternalWithRetry()218 freed_by_count = (*allocation_attr.freed_by_func)(); in AllocateRawInternalWithRetry()235 if (allocation_attr.freed_by_func != nullptr) { in AllocateRaw()236 freed_by_count = (*allocation_attr.freed_by_func)(); in AllocateRaw()
286 std::function<uint64()> freed_by_func = [dst_device, &safe_alloc_frontier]() { in SameWorkerRecvDone() local291 allocation_attr.freed_by_func = &freed_by_func; in SameWorkerRecvDone()
593 std::function<uint64()> freed_by_func = [this, &safe_alloc_frontier]() { in MaybeCopyTensorToGPU() local598 allocation_attr.freed_by_func = &freed_by_func; in MaybeCopyTensorToGPU()