| .. |
|
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
|
Add groups to Conv1d (#948)
|
2024-04-27 06:24:57 -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
|
Added ArcTan2 operation (#1079)
|
2024-05-08 08:35:15 -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
|
Added ArcTan2 operation (#1079)
|
2024-05-08 08:35:15 -07:00 |
|
primitives.cpp
|
std and expm1 (#973)
|
2024-04-08 14:26:01 -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 |
cc05a281c4