Add MNIST 2 version (with batch normalization) in C - Githubissues

certik / mlc

Machine Learning Compiler

MIT License

9 stars 1 forks source link

Add MNIST 2 version (with batch normalization) in C #81

Closed certik closed 5 months ago

certik commented 5 months ago

TODO:

[x] Implement batchnorm2d kernel
[x] Get everything running
[x] refactor into separate files and CI tests