update rnncell synops calculation

NeuroBench / neurobench

Benchmark harness and baseline results for the NeuroBench algorithm track.

https://neurobench.readthedocs.io

Apache License 2.0

52 stars 12 forks source link

Closed jasonlyik closed 1 month ago

jasonlyik commented 1 month ago

Updating LSTMCell and GRUCell synops calculation,

Previously the reshape of ifgo/rzn for the LSTM/GRU blocks assumed batch=1 and flattened the hidden features.

Changed to arbitrary batch size.

No difference in output for the unittests or LSTM example for mackey_glass.

jasonlyik commented 1 month ago

@korneelf1 Can you check over these changes to see if they look correct?