Searched defs:alloc_id (Results 1 – 5 of 5) sorted by relevance
63 uint64_t ScrambleAllocId(uint64_t alloc_id, uint32_t thread_idx) { in ScrambleAllocId()69 uint64_t alloc_id = 0; in Thread() local
330 for (int64_t alloc_id : memory_stats.persistent_tensor_alloc_ids()) { in RecordMemoryStats() local353 int64_t alloc_id) { in RecordAllocationId()515 int64_t alloc_id = AllocationId(n, i); in AddToCostGraphDef() local
135 let alloc_id = resources.get_anon_alloc(); in allocate_regions() localVariable
155 for (const auto& alloc_id : ctx->persistent_alloc_ids()) { in SetMemory() local
992 int64_t alloc_id) { in record_persistent_memory_allocation()