CINN
CINN copied to clipboard
Compiler Infrastructure for Neural Networks
实现batch_norm_train的单测,结果和Paddle对比。
This PR did 3 things: 1. Add conv2d_winograd compute and schedule 2. Fix Fuse schedule 3. Enhance simplify: For examples: a = 2x and b = 8 -> a/b =...
Test ci
test ci
This pr added performance tests in model tests.
This pr supports 2 new Models: `MobileNetV1` and `ResNet50`. And their corresponding tests are added, too. To achieve this, a new op `reshape2 ` is added and some bugs are...
增加`ConvertToMoreFusion` Pass处理Graph中潜在的融合可能,该Pass将修改Graph的结构。