Max-Heinrich Laves ff4223904d Conv3d (#993) ...

* added conv3d

added conv3d

implemented explicit_gemm_conv_ND_cpu and bounds checks for slow_conv_3D

* incorporated reviewer comments

* fixed test

* reduced tensor shapes in test for conv3d

* Reviewer suggestion

Co-authored-by: Awni Hannun <awni.hannun@gmail.com>

Reviewer suggestion

Co-authored-by: Awni Hannun <awni.hannun@gmail.com>

Reviewer suggestion

Co-authored-by: Awni Hannun <awni.hannun@gmail.com>

Reviewer suggestion

2024-05-11 06:15:02 -07:00

..

mlx

Conv3d (#993)

2024-05-11 06:15:02 -07:00

src

Conv3d (#993)

2024-05-11 06:15:02 -07:00

tests

Conv3d (#993)

2024-05-11 06:15:02 -07:00