..
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
Ios compile ( #784 )
2024-03-04 20:02:26 -08:00
compiled_cpu.cpp
Ios compile ( #784 )
2024-03-04 20:02:26 -08: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
Implement the 'where' primitive for conditional selection ( #664 )
2024-02-22 15:10:48 -08: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
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
Refactor CPU compile preamble ( #708 )
2024-02-19 06:12:53 -08: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
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
copyright + ack
2023-11-30 11:12:53 -08:00