Angelos Katharopoulos 29d0c10ee5 Reshape improvement (#818)

2024-03-12 17:54:31 -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

Refactor CPU compile preamble (#708)

2024-02-19 06:12:53 -08:00

binary.h

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

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

CMakeLists.txt

Add SVD primitive (#809)

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

compiled_cpu.cpp

Compile stride bug (#812)

2024-03-11 06:31:31 -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

Ios compile (#784)

2024-03-04 20:02:26 -08:00

compiled.h

Shapeless compilation for some graphs (#687)

2024-02-19 21:43:54 -08:00

conv.cpp

Convolution update (#651)

2024-02-28 20:11:16 -08:00

copy.cpp

Buffer Donation (#519)

2024-01-26 16:30:33 -08:00

copy.h

copyright + ack

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

default_primitives.cpp

Add SVD primitive (#809)

2024-03-12 12:30:11 -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

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

Refactor CPU compile preamble (#708)

2024-02-19 06:12:53 -08:00

ops.h

Fix logsumexp edge case (#740)

2024-02-25 08:39:55 -08:00

primitives.cpp

Reshape improvement (#818)

2024-03-12 17:54:31 -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

Refactor the reduction kernels (#277)

2023-12-24 14:47:57 -08:00

rope.cpp

Separate fast ops and primitives (#699)

2024-02-16 19:16:39 -08:00

scan.cpp

copyright + ack

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

select.cpp

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

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

softmax.cpp

Fix the softmax fix (#661)

2024-02-09 17:02:13 -08:00

sort.cpp

copyright + ack

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

svd.cpp

Add SVD primitive (#809)

2024-03-12 12:30:11 -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

Reshape improvement (#818)

2024-03-12 17:54:31 -07:00