LSTM and LSTM gradient implementation for training #15034
baijumeswani
merged
22 commits into
main
from
baijumeswani/lstm-grad
Schema changes for LSTMInternal and LSTMGrad
f69eb1c3
LSTMInternal forward graph execution implementation
8dc3045f
LSTMGrad gradient graph execution implementation
ef139d22
LSTMInternal and LSTMGrad cpu kernel registration
0dee1c09
Rule based graph transformer to replace LSTM node with LSTMInternal node
909cf50e
Gradient building for LSTMInternal
e73abe18
Graph transformer test for LSTMReplacement
f876e1b6
LSTMInternal and LSTMGrad computation tests
c30ed416
Correct LSTM gradient building typo
63db9d9b
Address build failures
55559d72
Address build failures
d3983562
Fix windows build errors
326090ef
Change name of LSTMInternal to LSTMTraining
62fee513
Combine training and shared uni directional lstm
eaeee2f1
Address PR comments, SAL errors, and lint errors
2e2dcef9
Resolving SAL errors and pylint warnings
537c1d2a
Merge branch 'main' of https://github.com/microsoft/onnxruntime into …
b2f37f43
pengwa
commented
on 2023-03-17
Address pull request review comments
7f45ae98
Merge branch 'main' of https://github.com/microsoft/onnxruntime into …
f40394d6
fix unit test
dc9270b8
Address pull request review comments
35e93b85
Merge branch 'main' of https://github.com/microsoft/onnxruntime into …
27f27148
pengwa
approved these changes
on 2023-03-22
Login to write a write a comment.
Login via GitHub