DenseKernel |
defprotocol |
lib/fib/tensor/linalg.fib:63 |
Element |
defprotocol |
lib/fib/tensor/storage.fib:6 |
MatrixKernel |
defprotocol |
lib/fib/tensor/linalg.fib:41 |
Optimiser |
defprotocol |
lib/fib/tensor/optim.fib:139 |
OptimiserOwned |
defprotocol |
lib/fib/tensor/optim-owned.fib:121 |
PackedReduction |
defprotocol |
lib/fib/tensor/simd.fib:5 |
Rows |
defprotocol |
lib/fib/tensor/vmath.fib:312 |
Transcendental |
defprotocol |
lib/fib/tensor/vmath.fib:142 |
UnaryLanes |
defprotocol |
lib/fib/tensor/unary.fib:51 |
abs |
defun |
lib/fib/tensor/unary.fib:58 |
activate |
defun |
lib/fib/tensor/vmath.fib:151 |
activate8 |
defun |
lib/fib/tensor/vmath.fib:113 |
activation-code |
defun |
lib/fib/tensor/vmath.fib:95 |
adam-step |
defun |
lib/fib/tensor/optim.fib:149 |
adam-step-owned |
defun |
lib/fib/tensor/optim-owned.fib:138 |
add |
defun |
lib/fib/tensor/elementwise.fib:41 |
all? |
defun |
lib/fib/tensor/masks.fib:63 |
any? |
defun |
lib/fib/tensor/masks.fib:60 |
arange |
defun |
lib/fib/tensor/storage.fib:110 |
argmax |
defun |
lib/fib/tensor/reduce-extra.fib:53 |
argmin |
defun |
lib/fib/tensor/reduce-extra.fib:54 |
assoc |
defun |
lib/fib/tensor/storage.fib:105 |
at |
defun |
lib/fib/tensor/storage.fib:36 |
axis-shape |
defun |
lib/fib/tensor/reduce.fib:34 |
axpby |
defun |
lib/fib/tensor/elementwise.fib:67 |
broadcast-to |
defun |
lib/fib/tensor/views.fib:41 |
checked-size |
defun |
lib/fib/tensor/storage.fib:46 |
contiguous |
defun |
lib/fib/tensor/storage.fib:97 |
contiguous? |
defun |
lib/fib/tensor/storage.fib:30 |
copy |
defun |
lib/fib/tensor/storage.fib:84 |
dense |
defun |
lib/fib/tensor/linalg.fib:71 |
div |
defun |
lib/fib/tensor/elementwise.fib:51 |
dot |
defun |
lib/fib/tensor/linalg.fib:7 |
dot-fast |
defun |
lib/fib/tensor/simd.fib:17 |
dtype |
defun |
lib/fib/tensor/storage.fib:26 |
equal |
defun |
lib/fib/tensor/masks.fib:22 |
exp |
defun |
lib/fib/tensor/vmath.fib:147 |
flat-at |
defun |
lib/fib/tensor/storage.fib:39 |
flip |
defun |
lib/fib/tensor/views.fib:24 |
fold |
defun |
lib/fib/tensor/reduce.fib:7 |
fold-axis |
defun |
lib/fib/tensor/reduce.fib:59 |
fold-planes |
defun |
lib/fib/tensor/reduce.fib:51 |
from-array |
defun |
lib/fib/tensor/storage.fib:21 |
from-parts |
defun |
lib/fib/tensor/storage.fib:18 |
from-vec |
defun |
lib/fib/tensor/storage.fib:60 |
full |
defun |
lib/fib/tensor/storage.fib:50 |
greater |
defun |
lib/fib/tensor/masks.fib:28 |
greater-equal |
defun |
lib/fib/tensor/masks.fib:30 |
item |
defun |
lib/fib/tensor/storage.fib:42 |
layernorm |
defun |
lib/fib/tensor/vmath.fib:322 |
less |
defun |
lib/fib/tensor/masks.fib:24 |
less-equal |
defun |
lib/fib/tensor/masks.fib:26 |
linspace |
defun |
lib/fib/tensor/storage.fib:120 |
log |
defun |
lib/fib/tensor/vmath.fib:148 |
map |
defun |
lib/fib/tensor/elementwise.fib:20 |
map-as |
defun |
lib/fib/tensor/elementwise.fib:9 |
maximum |
defun |
lib/fib/tensor/reduce.fib:22 |
maximum-axis |
defun |
lib/fib/tensor/reduce-extra.fib:32 |
mean |
defun |
lib/fib/tensor/reduce.fib:30 |
mean-axis |
defun |
lib/fib/tensor/reduce-extra.fib:37 |
mean-fast |
defun |
lib/fib/tensor/simd.fib:22 |
minimum |
defun |
lib/fib/tensor/reduce.fib:26 |
minimum-axis |
defun |
lib/fib/tensor/reduce-extra.fib:34 |
mmul |
defun |
lib/fib/tensor/linalg.fib:52 |
mmul-bias |
defun |
lib/fib/tensor/linalg.fib:74 |
mmul-scalar |
defun |
lib/fib/tensor/linalg.fib:22 |
mul |
defun |
lib/fib/tensor/elementwise.fib:47 |
neg |
defun |
lib/fib/tensor/elementwise.fib:60 |
ones |
defun |
lib/fib/tensor/storage.fib:54 |
outer |
defun |
lib/fib/tensor/linalg.fib:14 |
permute |
defun |
lib/fib/tensor/views.fib:9 |
plane-bases |
defun |
lib/fib/tensor/reduce.fib:45 |
prod |
defun |
lib/fib/tensor/reduce.fib:19 |
prod-axis |
defun |
lib/fib/tensor/reduce-extra.fib:5 |
rank |
defun |
lib/fib/tensor/storage.fib:27 |
relu |
defun |
lib/fib/tensor/vmath.fib:153 |
relu-grad |
defun |
lib/fib/tensor/masks.fib:51 |
relu-mask |
defun |
lib/fib/tensor/masks.fib:56 |
reshape |
defun |
lib/fib/tensor/views.fib:4 |
rsqrt |
defun |
lib/fib/tensor/unary.fib:57 |
scale |
defun |
lib/fib/tensor/elementwise.fib:54 |
sgd-momentum-step |
defun |
lib/fib/tensor/optim.fib:153 |
shape |
defun |
lib/fib/tensor/storage.fib:25 |
shift |
defun |
lib/fib/tensor/elementwise.fib:57 |
sign |
defun |
lib/fib/tensor/unary.fib:59 |
size |
defun |
lib/fib/tensor/storage.fib:28 |
slice |
defun |
lib/fib/tensor/views.fib:31 |
softmax |
defun |
lib/fib/tensor/vmath.fib:319 |
sqrt |
defun |
lib/fib/tensor/unary.fib:56 |
square |
defun |
lib/fib/tensor/elementwise.fib:63 |
strides |
defun |
lib/fib/tensor/storage.fib:29 |
sub |
defun |
lib/fib/tensor/elementwise.fib:44 |
sum |
defun |
lib/fib/tensor/reduce.fib:16 |
sum-axis |
defun |
lib/fib/tensor/reduce.fib:81 |
sum-fast |
defun |
lib/fib/tensor/simd.fib:15 |
tanh |
defun |
lib/fib/tensor/vmath.fib:149 |
tensor |
defun |
lib/fib/tensor/storage.fib:66 |
tensor-i64 |
defun |
lib/fib/tensor/storage.fib:67 |
to-array |
defun |
lib/fib/tensor/storage.fib:101 |
to-vec |
defun |
lib/fib/tensor/storage.fib:69 |
transpose |
defun |
lib/fib/tensor/views.fib:19 |
validate |
defun |
lib/fib/tensor/storage.fib:33 |
where |
defun |
lib/fib/tensor/masks.fib:33 |
zeros |
defun |
lib/fib/tensor/storage.fib:53 |
zeros-f32 |
defun |
lib/fib/tensor/storage.fib:55 |
zeros-i32 |
defun |
lib/fib/tensor/storage.fib:56 |
zeros-i64 |
defun |
lib/fib/tensor/storage.fib:57 |
zip-with |
defun |
lib/fib/tensor/elementwise.fib:23 |