issues
search
jiazhihao
/
TASO
The Tensor Algebra SuperOptimizer for Deep Learning
Apache License 2.0
682
stars
89
forks
source link
Bug fixes to correct import and export of ONNX models
#64
Closed
santhnm2
closed
4 years ago
santhnm2
commented
4 years ago
Summary of fixes:
Add Flatten and GlobalAveragePooling ops
Import biases for Conv/GeMM operators
Only transpose GeMM inputs if explicitly requested
Skip transpose when preprocessing weights for final graph
Add epsilon attribute to BatchNorm
Ensure Conv padding is same on both sides
Matmul -> MatMul for ONNX compatibility
Summary of fixes: