search

pulp-platform / dory

A tool to deploy Deep Neural Networks on PULP-based SoC's
Apache License 2.0
72 stars 22 forks source link

Partial fix for Diana_TVM when layer dimensions are not a multiple of 16 #60

Closed maartenvds closed 9 months ago

maartenvds commented 9 months ago

Input/output channels don't need to be multiple of 16 anymore. Still need to fix this for depthwise convolutions.

For the width however, additional padding is needed -> WIP.

Changes were proposed by Amine and tested by myself.