RNNprofiler: fix gates size retrieval logic in _rnn_flops #3921
tjruwase
merged
6 commits into
deepspeedai:master
from
pinstripe-potoroo/fix-rnn-flop-profiler-gates-size-logic
fix gates size retrieval logic in _rnn_flops
a4c60774
Merge branch 'master' into pinstripe-potoroo/fix-rnn-flop-profiler-ga…
d291aeb7
Merge branch 'master' into pinstripe-potoroo/fix-rnn-flop-profiler-ga…
3cbc4094
Merge branch 'master' into pinstripe-potoroo/fix-rnn-flop-profiler-ga…
0acbe66f
Merge branch 'master' into pinstripe-potoroo/fix-rnn-flop-profiler-ga…
1fb46571
Merge branch 'master' into pinstripe-potoroo/fix-rnn-flop-profiler-ga…
1e11dfe1
tjruwase
approved these changes
on 2023-07-24
Login to write a write a comment.
Login via GitHub