diff options
author | Fredrik Svedberg <fredrik.svedberg@arm.com> | 2021-05-11 13:51:47 +0200 |
---|---|---|
committer | Mauricio Briceno <mauricio.briceno@arm.com> | 2021-05-13 18:01:21 +0200 |
commit | 93d5c35feaae8f2fe8e0361ffc19e31d463ca099 (patch) | |
tree | c5179f8e06c5aa3e16513849f4be338393661afb /ethosu/mlw_codec/mlw_encode.h | |
parent | 67e11f7bce40d72e0dda97cf658a3c3ee600c1eb (diff) | |
download | ethos-u-vela-93d5c35feaae8f2fe8e0361ffc19e31d463ca099.tar.gz |
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
Diffstat (limited to 'ethosu/mlw_codec/mlw_encode.h')
-rw-r--r-- | ethosu/mlw_codec/mlw_encode.h | 20 |
1 files changed, 20 insertions, 0 deletions
diff --git a/ethosu/mlw_codec/mlw_encode.h b/ethosu/mlw_codec/mlw_encode.h index a995ac6e..743603b5 100644 --- a/ethosu/mlw_codec/mlw_encode.h +++ b/ethosu/mlw_codec/mlw_encode.h @@ -38,6 +38,26 @@ int mlw_encode(int16_t *inbuf, int inbuf_size, uint8_t **outbuf, int verbose); 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 |