Skip to content

Fix profiling for GRU/LSTM

Javier Duarte requested to merge github/fork/drankincms/fix_rnn_profiling into main

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

Merge request reports

Loading