mlx/python/mlx/nn
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
..
layers Transposed Convolution (#1245) 2024-09-06 19:52:38 -07:00
__init__.py Common neural network initializers nn.initializers (#456) 2024-01-23 06:47:20 -08:00
init.py Some fixes to typing (#1371) 2024-08-28 11:16:19 -07:00
losses.py Some fixes to typing (#1371) 2024-08-28 11:16:19 -07:00
utils.py Some fixes to typing (#1371) 2024-08-28 11:16:19 -07:00