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
..
layers	Conv3d (#993)	2024-05-11 06:15:02 -07:00
__init__.py	Common neural network initializers nn.initializers (#456)	2024-01-23 06:47:20 -08:00
init.py	post nanobind docs fixes and some updates (#889)	2024-03-24 15:03:27 -07:00
losses.py	post nanobind docs fixes and some updates (#889)	2024-03-24 15:03:27 -07:00
utils.py	Custom VJP and checkpointing (#541)	2024-01-30 16:04:45 -08:00