add dummy allocate_tensors() function to match tf.lite.Interpreter API

xmos · Aug 16, 2023 · b0c2f19 · b0c2f19
1 parent 19ad2ac
commit b0c2f19
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/python/xmos_ai_tools/xinterpreters/base/base_interpreter.py b/python/xmos_ai_tools/xinterpreters/base/base_interpreter.py
@@ -103,6 +103,10 @@ def print_memory_plan(self) -> None:
         """! Abstract method to print a plan of memory allocation"""
         raise NotImplementedError
 
+    def allocate_tensors(self):
+        """! Dummy function to match tf.lite.Interpreter() API"""
+        return
+
     def get_input_tensor_size(self, input_index: int = 0, model_index: int = 0) -> int:
         """! Read the size of the input tensor from the model.
         @param input_index  The index of input tensor to target.