| 41 | ArenaPlanner::~ArenaPlanner() {} |
| 42 | |
| 43 | int64_t ArenaPlanner::BasePointer(TfLiteAllocationType type) { |
| 44 | if (type == kTfLiteArenaRwPersistent) { |
| 45 | return persistent_arena_.BasePointer(); |
| 46 | } |
| 47 | if (type == kTfLiteArenaRw) { |
| 48 | return arena_.BasePointer(); |
| 49 | } |
| 50 | return 0; |
| 51 | } |
| 52 | |
| 53 | TfLiteStatus ArenaPlanner::ResetAllocations() { |
| 54 | TF_LITE_ENSURE_STATUS(arena_.Clear()); |