linear_memory_planner.h

00001 /* Copyright 2019 The TensorFlow Authors. All Rights Reserved.
00002 
00003 Licensed under the Apache License, Version 2.0 (the "License");
00004 you may not use this file except in compliance with the License.
00005 You may obtain a copy of the License at
00006 
00007     http://www.apache.org/licenses/LICENSE-2.0
00008 
00009 Unless required by applicable law or agreed to in writing, software
00010 distributed under the License is distributed on an "AS IS" BASIS,
00011 WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
00012 See the License for the specific language governing permissions and
00013 limitations under the License.
00014 ==============================================================================*/
00015 
00016 #ifndef TENSORFLOW_LITE_EXPERIMENTAL_MICRO_MEMORY_PLANNER_LINEAR_MEMORY_PLANNER_H_
00017 #define TENSORFLOW_LITE_EXPERIMENTAL_MICRO_MEMORY_PLANNER_LINEAR_MEMORY_PLANNER_H_
00018 
00019 #include "tensorflow/lite/experimental/micro/memory_planner/memory_planner.h"
00020 
00021 namespace tflite {
00022 
00023 // The simplest possible memory planner that just lays out all buffers at
00024 // increasing offsets without trying to reuse memory.
00025 class LinearMemoryPlanner : public MemoryPlanner {
00026  public:
00027   LinearMemoryPlanner();
00028   ~LinearMemoryPlanner() override;
00029 
00030   TfLiteStatus AddBuffer(tflite::ErrorReporter* error_reporter, int size,
00031                          int first_time_used, int last_time_used) override;
00032 
00033   int GetMaximumMemorySize() override;
00034   int GetBufferCount() override;
00035   TfLiteStatus GetOffsetForBuffer(tflite::ErrorReporter* error_reporter,
00036                                   int buffer_index, int* offset) override;
00037 
00038  private:
00039   static constexpr int kMaxBufferCount = 1024;
00040   int buffer_offsets_[kMaxBufferCount];
00041   int current_buffer_count_;
00042   int next_free_offset_;
00043 };
00044 
00045 }  // namespace tflite
00046 
00047 #endif  // TENSORFLOW_LITE_EXPERIMENTAL_MICRO_MEMORY_PLANNER_LINEAR_MEMORY_PLANNER_H_