zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-12-16 01:49:05 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity
Files
b6d643fc17d6ae769dc005492141224836ba00d2
mlx/python/tests/test_conv.py
Jagrit Digani 8777fd104f Depthwise Conv2D optimization (#2036) 
- Add new specialized kernel for small kernel (kernels size <= 7), small strides (strides <= 2) depthwise 2d convolutions
- Add related tests
2025-04-03 09:42:04 -07:00

36 KiB
Raw Blame History

View Raw
Reference in New Issue View Git Blame Copy Permalink