| .. |
|
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
|
Fast RMS Norm (#862)
|
2024-03-21 07:20:54 -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
|
Add types and order in kernel name (#831)
|
2024-03-13 20:34:06 -07: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
|
Add a SliceUpdate op and primitive (#850)
|
2024-03-20 10:39:25 -07:00 |
|
copy.h
|
Add a SliceUpdate op and primitive (#850)
|
2024-03-20 10:39:25 -07:00 |
|
default_primitives.cpp
|
Add a SliceUpdate op and primitive (#850)
|
2024-03-20 10:39:25 -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
|
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
|
Add a SliceUpdate op and primitive (#850)
|
2024-03-20 10:39:25 -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 |
|
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
|
Fast RMS Norm (#862)
|
2024-03-21 07:20:54 -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 a SliceUpdate op and primitive (#850)
|
2024-03-20 10:39:25 -07:00 |
