From b91adbb9f2ede51e1ccdeae80b4db248ee89adc7 Mon Sep 17 00:00:00 2001 From: jihandong Date: Tue, 28 May 2024 16:24:21 +0800 Subject: [PATCH] ml: useful tflm debug options print memory plan, and time cost of each operators. Signed-off-by: jihandong --- mlearning/tflite-micro/Makefile | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/mlearning/tflite-micro/Makefile b/mlearning/tflite-micro/Makefile index 1db63a28b..eb03470f4 100644 --- a/mlearning/tflite-micro/Makefile +++ b/mlearning/tflite-micro/Makefile @@ -55,7 +55,10 @@ COMMON_FLAGS += -Wno-undef COMMON_FLAGS += -Wno-unused-variable COMMON_FLAGS += -DTFLITE_EMULATE_FLOAT COMMON_FLAGS += -DTF_LITE_DISABLE_X86_NEON -ifeq ($(CONFIG_TFLITEMICRO_DEBUG),) +ifneq ($(CONFIG_TFLITEMICRO_DEBUG),) +COMMON_FLAGS += -DTF_LITE_SHOW_MEMORY_USE +COMMON_FLAGS += -DTF_LITE_USE_CTIME +else COMMON_FLAGS += -DTF_LITE_STRIP_ERROR_STRINGS endif