Fix profiling for GRU/LSTM
Created by: drankincms
A# Description
This PR addresses #829 by fixing the profiling tool for GRU and LSTM layers.
Type of change
For a new feature or function, please create an issue first to discuss it with us before submitting a pull request.
-
Bug fix (non-breaking change that fixes an issue)
Tests
Tested with the script from #829 .
Checklist
-
I have read the guidelines for contributing. -
My changes generate no new warnings.