From 74471527670e0cb52a1003ad283a98b5eaed8d46 Mon Sep 17 00:00:00 2001
From: Jiang Yufan <jiang0426@sjtu.edu.cn>
Date: Tue, 22 Aug 2023 14:05:10 +0800
Subject: [PATCH] Tensorflow test update

---
 APP_Framework/Applications/knowing_app/mnist/main.cpp    | 2 +-
 APP_Framework/Applications/knowing_app/mnist/mnistmain.c | 1 +
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/APP_Framework/Applications/knowing_app/mnist/main.cpp b/APP_Framework/Applications/knowing_app/mnist/main.cpp
index 3f2d3321a..c8c264490 100644
--- a/APP_Framework/Applications/knowing_app/mnist/main.cpp
+++ b/APP_Framework/Applications/knowing_app/mnist/main.cpp
@@ -1,5 +1,5 @@
 #include <cstdio>
-#include <transform.h>
+// #include <transform.h>
 #include <stdio.h>
 
 #include "tensorflow/lite/micro/all_ops_resolver.h"
diff --git a/APP_Framework/Applications/knowing_app/mnist/mnistmain.c b/APP_Framework/Applications/knowing_app/mnist/mnistmain.c
index fa1c26821..b0f647d49 100644
--- a/APP_Framework/Applications/knowing_app/mnist/mnistmain.c
+++ b/APP_Framework/Applications/knowing_app/mnist/mnistmain.c
@@ -26,3 +26,4 @@ void mnist_app(void);
 int tfmnist(void) {
     mnist_app();
 }
+PRIV_SHELL_CMD_FUNCTION(tfmnist, a tenorflow_lite_for_microcontroller sample, PRIV_SHELL_CMD_FUNC_ATTR);