3 Commits

Author	SHA1	Message	Date
Nikhil Mehta	0b7d71fd2f	Add softmin, hardshrink, hardtanh (#1180) ... --------- Co-authored-by: Nikhil Mehta <nikmehta@tesla.com>	2024-06-04 15:48:18 -07:00
Rifur13	9401507336	Add groups to 2-D convolutions (#1129) ... * Added groups to 2-D convolutions. Only implemented for **some** specializations. Also fixed 1D grouped convs with different kernel strides and added more tests. * fix channels condition	2024-05-22 20:01:44 -07:00
Jagrit Digani	776c3d226d	Convolution update (#651) ... * Init steel conv and update Conv primitive * Update slow CPU implementation to support flipping and input dilation winograd conv routing Co-authored-by: Awni Hannun <awni@apple.com>	2024-02-28 20:11:16 -08:00