Max-Heinrich Laves efeb9c0f02 Transposed Convolution (#1245) ... * initial implementation for conv_transpose ran pre-commit implemented conv_transpose updated conv_general docstring updated conv_general docstring updated code comments removed commented run_conv_checks updated acknowledgments added missing entry to ops.rst added op to nn.layers resolved merge conflicts * removed ConvolutionTranspose primitive as suggested by reviewer removed ConvolutionTranspose primitive as suggested by reviewer * remove transpose flag, add another test --------- Co-authored-by: Awni Hannun <awni@apple.com>		2024-09-06 19:52:38 -07:00
..
arange.h	copyright + ack	2023-11-30 11:12:53 -08:00
arg_reduce.cpp	copyright + ack	2023-11-30 11:12:53 -08:00
binary_two.h	Implement the 'where' primitive for conditional selection (#664)	2024-02-22 15:10:48 -08:00
binary.cpp	smaller CPU binary (#1203)	2024-06-14 09:46:55 -07:00
binary.h	JIT compile option for binary minimization (#1091)	2024-05-22 12:57:13 -07:00
cholesky.cpp	JIT compile option for binary minimization (#1091)	2024-05-22 12:57:13 -07:00
CMakeLists.txt	Fast Hadamard Transform (#1249)	2024-07-09 20:39:01 -07:00
common.cpp	Custom transforms (#1246)	2024-07-10 18:00:01 -07:00
compiled_cpu.cpp	Read arrays from files faster (#1330)	2024-08-14 20:09:56 -07:00
compiled_nocpu.cpp	Ios compile (#784)	2024-03-04 20:02:26 -08:00
compiled_preamble.h	Refactor CPU compile preamble (#708)	2024-02-19 06:12:53 -08:00
compiled.cpp	Fix compiel with byte sized constants (#1381)	2024-08-30 17:24:35 -07:00
compiled.h	Fix cpu compile (#934)	2024-04-01 17:37:12 -07:00
conv.cpp	Transposed Convolution (#1245)	2024-09-06 19:52:38 -07:00
copy.cpp	Fix reshape copy bug (#1253)	2024-07-07 21:37:00 -07:00
copy.h	Add a SliceUpdate op and primitive (#850)	2024-03-20 10:39:25 -07:00
default_primitives.cpp	Custom transforms (#1246)	2024-07-10 18:00:01 -07:00
erf.cpp	copyright + ack	2023-11-30 11:12:53 -08:00
fft.cpp	copyright + ack	2023-11-30 11:12:53 -08:00
hadamard.cpp	Fast Hadamard Transform (#1249)	2024-07-09 20:39:01 -07:00
hadamard.h	Fast Hadamard Transform (#1249)	2024-07-09 20:39:01 -07:00
indexing.cpp	copyright + ack	2023-11-30 11:12:53 -08:00
inverse.cpp	CPU mx.linalg.cholesky_inverse and mx.linalg.tri_inv (#1307)	2024-08-08 15:18:02 -07:00
lapack_helper.h	Add SVD primitive (#809)	2024-03-12 12:30:11 -07:00
load.cpp	Async load (#1372)	2024-08-28 14:21:55 -07:00
load.h	Async load (#1372)	2024-08-28 14:21:55 -07:00
make_compiled_preamble.sh	fix compiling with space in paths (#1332)	2024-08-15 16:39:24 -07:00
masked_mm.cpp	Float mask update (#1152)	2024-05-23 17:20:44 -07:00
ops.h	GPU mx.sign for complex64 (#1326)	2024-08-14 07:54:53 -07:00
primitives.cpp	Fix slice data size (#1394)	2024-09-04 19:10:43 -07:00
qrf.cpp	fixes for building and running on iOS (#619)	2024-02-04 12:29:17 -08:00
quantized.cpp	Rename block sparse (#1149)	2024-05-22 07:48:34 -07:00
reduce_utils.cpp	Fix reduce edge case (#1389)	2024-09-01 21:37:51 -07:00
reduce.cpp	Add isfinite (#1318)	2024-08-13 14:49:28 -07:00
reduce.h	Refactor reductions and fix scatter atomics for large sizes (#1300)	2024-08-22 16:03:31 -07:00
scan.cpp	Fix scan (#1188)	2024-06-05 14:21:58 -07:00
select.cpp	Implement the 'where' primitive for conditional selection (#664)	2024-02-22 15:10:48 -08:00
slicing.cpp	Fix slice data size (#1394)	2024-09-04 19:10:43 -07:00
slicing.h	Fix slice data size (#1394)	2024-09-04 19:10:43 -07:00
softmax.cpp	Option for precise softmax (#953)	2024-04-04 08:32:35 -07:00
sort.cpp	Fix strided sort bug (#1236)	2024-06-26 14:32:11 -07:00
svd.cpp	No CPU option for binary minimization (#1105)	2024-05-13 16:08:11 -07:00
ternary.h	Fix ternary for large arrays (#1359)	2024-08-26 11:22:27 -07:00
threefry.cpp	copyright + ack	2023-11-30 11:12:53 -08:00
threefry.h	copyright + ack	2023-11-30 11:12:53 -08:00
unary.h	Refactor CPU compile preamble (#708)	2024-02-19 06:12:53 -08:00
utils.h	Fix slice data size (#1394)	2024-09-04 19:10:43 -07:00