zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-06-29 13:01:14 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity
2c7e9b5158
mlx/python/src
History
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