Function AddUsageRecord

tensorflow/lite/delegates/gpu/gl/runtime.cc:401–426

We assume that AddUsageRecord for different objects is called in order of program_id.

// We assume that AddUsageRecord for different objects is called in order of
// program_id.
Status AddUsageRecord(CombinedUsageRecords* usage_records, const Object& object,
                      const size_t program_id) {
  auto ref = GetRef(object);
  if (ref >= usage_records->usage_refs.size()) {
    usage_records->usage_refs.resize(ref + 1, kNotAssigned);
  }
  auto& usage_ref = usage_records->usage_refs[ref];
  if (object.object_type == ObjectType::BUFFER) {
    if (usage_ref == kNotAssigned) {
      usage_ref = usage_records->buffers.size();
      usage_records->buffers.emplace_back(
          /*tensor_size=*/NumElements(object.size),
          /*first_task=*/program_id,
          /*last_task=*/program_id);
    } else {
      UpdateUsageRecord(&usage_records->buffers[usage_ref], program_id);
    }
    return OkStatus();
  }
  if (object.object_type == ObjectType::TEXTURE) {
    absl::visit(AddUsageRecordForTextureFunc{usage_records, ref, program_id},
                object.size);
    return OkStatus();
  }
  return InternalError("Unexpected object type");
}

Status ApplyBuffersAssignment(
    const ObjectsAssignment<size_t>& assignment,