uTensor / utensor_cgen

C++ code generator for uTensor https://utensor-cgen.readthedocs.io/en/latest/
Apache License 2.0
50 stars 40 forks source link

hotfix: reshape op data parser, inferring new shape from input tensors #130

Closed dboyliao closed 3 years ago

dboyliao commented 3 years ago

new shape sometime is input tensor in TFLite model file. That's why the tflite parser will fail randomly since the op options will be None in this case.

Screen Shot 2020-11-24 at 4 52 46 PM
victorromeo commented 3 years ago

Manually implementing this hot fix then running the unit tests in #133 shows this fix corrects the breaking test. See test_api/test_keras_tflite/test_export.py test_keras_model