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

..

array.cpp

Add conjugate operator (#1100)

2024-05-10 07:22:20 -07:00

buffer.h

Fix buffer protocol buffer size designation (#1010)

2024-04-19 06:06:13 -07:00

CMakeLists.txt

fix conversion to array (#1070)

2024-05-06 16:02:49 -07:00

constants.cpp

Switch to nanobind (#839)

2024-03-18 20:12:25 -07:00

convert.cpp

Shared events for synchronization + async eval (#998)

2024-04-17 06:16:02 -07:00

convert.h

Switch to nanobind (#839)

2024-03-18 20:12:25 -07:00

device.cpp

Switch to nanobind (#839)

2024-03-18 20:12:25 -07:00

fast.cpp

post nanobind docs fixes and some updates (#889)

2024-03-24 15:03:27 -07:00

fft.cpp

Switch to nanobind (#839)

2024-03-18 20:12:25 -07:00

indexing.cpp

fix slice update indexing (#1053)

2024-04-29 12:17:40 -07:00

indexing.h

Switch to nanobind (#839)

2024-03-18 20:12:25 -07:00

linalg.cpp

Switch to nanobind (#839)

2024-03-18 20:12:25 -07:00

load.cpp

Do not store iostream in shared_ptr (#872)

2024-03-22 06:54:45 -07:00

load.h

Switch to nanobind (#839)

2024-03-18 20:12:25 -07:00

metal.cpp

Reset peak memory (#1074)

2024-05-03 17:12:51 -07:00

mlx.cpp

Switch to nanobind (#839)

2024-03-18 20:12:25 -07:00

ops.cpp

Conv3d (#993)

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

random.cpp

Implementation of mlx.random.multivariate_normal (#502) (#877)

2024-04-09 13:50:12 -07:00

stream.cpp

Cpp docs (#1036)

2024-04-26 12:56:05 -07:00

transforms.cpp

add keyword positonal (#1081)

2024-05-06 07:18:49 -07:00

trees.cpp

Properly handle negative axes in python vmap (#944)

2024-04-02 18:07:23 -07:00

trees.h

Properly handle negative axes in python vmap (#944)

2024-04-02 18:07:23 -07:00

utils.cpp

fix conversion to array (#1070)

2024-05-06 16:02:49 -07:00

utils.h

fix conversion to array (#1070)

2024-05-06 16:02:49 -07:00