Searched defs:alloc_id (Results 1 – 4 of 4) sorted by relevance
327 for (int64 alloc_id : memory_stats.persistent_tensor_alloc_ids()) { in RecordMemoryStats() local350 int64 alloc_id) { in RecordAllocationId()511 int64 alloc_id = AllocationId(n, i); in AddToCostGraphDef() local
197 for (const auto& alloc_id : context->persistent_alloc_ids()) { in UpdateStats() local
147 for (const auto& alloc_id : ctx->persistent_alloc_ids()) { in SetMemory() local
775 int64 alloc_id = a->AllocationId(t->tensor_data().data()); in allocate_persistent() local901 int64 alloc_id) { in record_persistent_memory_allocation()