commit 93d5c35feaae8f2fe8e0361ffc19e31d463ca099
author Fredrik Svedberg <fredrik.svedberg@arm.com> Tue May 11 13:51:47 2021 +0200
committer Mauricio Briceno <mauricio.briceno@arm.com> Thu May 13 18:01:21 2021 +0200
tree c5179f8e06c5aa3e16513849f4be338393661afb
parent 67e11f7bce40d72e0dda97cf658a3c3ee600c1eb

Fix mlw_module

Fixedx size calculation in mlw_reorder_encode.
Fixed build warnings.

Signed-off-by: Fredrik Svedberg <fredrik.svedberg@arm.com>
Change-Id: Iac9408b9972a29b5a3403ba11f80dc4eaaa35453

ethosu/mlw_codec/mlw_encode.h

diff --git a/ethosu/mlw_codec/mlw_encode.h b/ethosu/mlw_codec/mlw_encode.h
index a995ac6..743603b 100644
--- a/ethosu/mlw_codec/mlw_encode.h
+++ b/ethosu/mlw_codec/mlw_encode.h
@@ -38,6 +38,26 @@
 EXPORTED
 void mlw_free_outbuf(uint8_t *outbuf);
 
+EXPORTED
+int mlw_reorder_encode(
+    int ifm_ublock_depth,
+    int ofm_ublock_depth,
+    int ofm_depth,
+    int kernel_height,
+    int kernel_width,
+    int ifm_depth,
+    int* brick_strides,
+    void* inbuf,
+    int ofm_block_depth,
+    int is_depthwise,
+    int is_partkernel,
+    int ifm_bitdepth,
+    int decomp_h,
+    int decomp_w,
+    uint8_t **outbuf,
+    int64_t* padded_length,
+    int verbose);
+
 #if __cplusplus
 }
 #endif