search

jiazhihao / TASO

The Tensor Algebra SuperOptimizer for Deep Learning
Apache License 2.0
692 stars 90 forks source link

Edit onnx export for PRelu #90

Closed ryujaehun closed 2 years ago

ryujaehun commented 2 years ago

Hi I fixed the code for PRelu export in TASO. It was confirmed that the bug occurred because previous code was partially modified.