majianjia / nnom

A higher-level Neural Network library for microcontrollers.
Apache License 2.0
815 stars 235 forks source link

about reshape #199

Open panhu opened 1 year ago

panhu commented 1 year ago

Hi: I want to ask what the nnom can support tf.reshape ?

majianjia commented 1 year ago

Could you check if this implementation is working? https://github.com/majianjia/nnom/blob/master/src/layers/nnom_reshape.c

panhu commented 1 year ago

Hi: I used a instead of b in my model, and when generating the. h file, it gave me "model. hook (reshappe_s (&tf. reshappe_12config), layer [1];", but there was no generat the other definitions of "tf. reshape 12_config" . and i used was "tf.reshape" not "tf.keras.layers.Reshape".

panhu commented 1 year ago

Hi: I used "tf.reshape" instead of "tf.keras.layers.Reshape" in my model, and when generating the. h file, it gave me "model. hook (reshappe_s (&tf. reshappe_12config), layer [1];", but there was no generat the other definitions of "tf. reshape 12_config" . and i used was "tf.reshape" not "tf.keras.layers.Reshape".