Merge branch 'fix-espnn-conv2d-incorrect-init' into 'master'

vikramdattu · vikramdattu · commit c65b8f9c383f · 2025-03-13T16:42:53.000+08:00
Fixed incorrect Init logic for ESP-NN Conv2D Kernel

See merge request app-frameworks/esp-tflite-micro!171
diff --git a/tensorflow/lite/micro/kernels/esp_nn/conv.cc b/tensorflow/lite/micro/kernels/esp_nn/conv.cc
@@ -45,7 +45,7 @@ struct NodeData {
 
 static void* Init(TfLiteContext* context, const char* buffer, size_t length) {
   TFLITE_DCHECK(context->AllocatePersistentBuffer != nullptr);
-  return context->AllocatePersistentBuffer(context, sizeof(OpDataConv));
+  return context->AllocatePersistentBuffer(context, sizeof(NodeData));
 }
 
 static TfLiteStatus Prepare(TfLiteContext* context, TfLiteNode* node) {

Original file line number	Diff line number	Diff line change
`@@ -45,7 +45,7 @@ struct NodeData {`
`45`	`45`
`46`	`46`	`static void* Init(TfLiteContext* context, const char* buffer, size_t length) {`
`47`	`47`	`TFLITE_DCHECK(context->AllocatePersistentBuffer != nullptr);`
`48`		`- return context->AllocatePersistentBuffer(context, sizeof(OpDataConv));`
	`48`	`+ return context->AllocatePersistentBuffer(context, sizeof(NodeData));`
`49`	`49`	`}`
`50`	`50`
`51`	`51`	`static TfLiteStatus Prepare(TfLiteContext* context, TfLiteNode* node) {`