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

..

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

Added ArcTan2 operation (#1079)

2024-05-08 08:35:15 -07:00

binary.h

Implement the 'where' primitive for conditional selection (#664)

2024-02-22 15:10:48 -08:00

CMakeLists.txt

Masked mm (#978)

2024-04-16 14:45:39 -07:00

compiled_cpu.cpp

Fix cpu compile (#934)

2024-04-01 17:37:12 -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 cpu compile (#934)

2024-04-01 17:37:12 -07:00

compiled.h

Fix cpu compile (#934)

2024-04-01 17:37:12 -07:00

conv.cpp

Conv3d (#993)

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

copy.cpp

Add missing && when forwarding args (#925)

2024-03-29 06:48:29 -07:00

copy.h

Add a SliceUpdate op and primitive (#850)

2024-03-20 10:39:25 -07:00

default_primitives.cpp

Add conjugate operator (#1100)

2024-05-10 07:22:20 -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

indexing.cpp

copyright + ack

2023-11-30 11:12:53 -08:00

inverse.cpp

Add vmap for SVD and inverse (#849)

2024-03-21 13:18:27 -07:00

lapack_helper.h

Add SVD primitive (#809)

2024-03-12 12:30:11 -07:00

load.cpp

Spelling (#342)

2024-01-01 21:08:17 -08:00

make_compiled_preamble.sh

Fix leak for multi-output primitives which are never detached (#1059)

2024-05-01 07:31:45 -07:00

masked_mm.cpp

Block sparse mm (#1058)

2024-05-02 14:03:58 -07:00

ops.h

Add conjugate operator (#1100)

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

primitives.cpp

Add conjugate operator (#1100)

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

qrf.cpp

fixes for building and running on iOS (#619)

2024-02-04 12:29:17 -08:00

quantized.cpp

Custom VJP and checkpointing (#541)

2024-01-30 16:04:45 -08:00

reduce.cpp

copyright + ack

2023-11-30 11:12:53 -08:00

reduce.h

Implement vjps for some primitives in the fast namespace (#883)

2024-03-26 16:35:34 -07:00

scan.cpp

add numeric type hierarchy and issubdtype as well as a set_dtype meth… (#427)

2024-03-25 12:32:59 -07:00

select.cpp

Implement the 'where' primitive for conditional selection (#664)

2024-02-22 15:10:48 -08:00

softmax.cpp

Option for precise softmax (#953)

2024-04-04 08:32:35 -07:00

sort.cpp

copyright + ack

2023-11-30 11:12:53 -08:00

svd.cpp

Add vmap for SVD and inverse (#849)

2024-03-21 13:18:27 -07:00

ternary.h

Implement the 'where' primitive for conditional selection (#664)

2024-02-22 15:10:48 -08: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

Add missing && when forwarding args (#894)

2024-03-25 14:55:54 -07:00