sony / nnabla

Neural Network Libraries
https://nnabla.org/
Apache License 2.0
2.73k stars 334 forks source link

fix tensor shape propagate error #1202

Closed TomonobuTsujikawa closed 1 year ago

TomonobuTsujikawa commented 1 year ago

TFLite exporter has a issue that the shape information (NCHW / NHWC) is not propagated correctly. This PR fix this.

Here is a example of this issue:

Before [-1,64,64,21] -> [-1,44,64,64] -> [-1,40,64,64]
After  [-1,64,64,21] -> [-1,64,64,44] -> [-1,64,64,40]