issues
search
ddkang
/
zkml
Apache License 2.0
348
stars
38
forks
source link
add python backwards converter [draft]
#11
Closed
punwai
closed
1 year ago
punwai
commented
1 year ago
Added new layers required for backprop, including permute, rotate (180 degrees), broadcast, and update.
Added a python script to transform forward pass computational graphs into back computational graph.
TODO:
Remove broadcast and write a generalized Avg2DPool backward pass
TODO: