Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/Narsil/fast_gpt2
/ functions
Functions
50 in github.com/Narsil/fast_gpt2
⨍
Functions
50
◇
Types & classes
22
↓ 7 callers
Method
forward
(&self, tensor: &mut OwnedTensor)
src/model/smelt.rs:177
↓ 6 callers
Function
download
( url: &str, filename: &str, max_files: usize, chunk_size: usize, )
src/download.rs:9
↓ 4 callers
Method
empty_past_key_values
(&self, dev: &Dev)
src/model/dfdx.rs:464
↓ 4 callers
Function
linear_from
( weight: TensorView, bias: TensorView, dev: &Dev, )
src/model/dfdx.rs:78
↓ 3 callers
Method
forward
(&self, tensor: FTensor<HiddenShape>)
src/model/dfdx.rs:251
↓ 3 callers
Function
layer_norm_from
( weight: TensorView, bias: TensorView, dev: &Dev, )
src/model/dfdx.rs:101
↓ 2 callers
Function
attention
( qkv: &T, qk: &mut TM, max: &mut [f32], past: &mut PastKeyValue, out: &mut OwnedTensor, )
src/model/smelt.rs:23
↓ 2 callers
Function
embedding_from
( weight: TensorView, dev: &Dev, )
src/model/dfdx.rs:112
↓ 2 callers
Function
split_qkv
( qkv: &T, past: &PastKeyValue, )
src/model/smelt.rs:80
↓ 1 callers
Function
attention_reshape
( qkv: &FTensor<QkvShape>, past_key: &FTensor<PastKeyShape>, past_value: &FTensor<PastValueShape>,
src/model/dfdx.rs:121
↓ 1 callers
Function
download_chunk
( client: reqwest::Client, url: String, filename: String, start: usize, stop: usize, )
src/download.rs:56
↓ 1 callers
Function
get_model
(filename: &str)
src/main.rs:32
↓ 1 callers
Function
get_tokenizer
(filename: &str)
src/main.rs:50
↓ 1 callers
Function
leak_buffer
(buffer: Mmap)
src/main.rs:26
↓ 1 callers
Function
run
()
src/lib.rs:39
↓ 1 callers
Function
special_argmax
Argmax of the last dimension of tensor `x `.
src/model/smelt.rs:454
↓ 1 callers
Function
unbiased_linear_from
( weight: TensorView, dev: &Dev, )
src/model/dfdx.rs:92
Function
attention_data
()
src/model/smelt.rs:602
Function
bench_add
(b: &mut Bencher)
benches/softmax.rs:42
Function
bench_addmm
(b: &mut Bencher)
benches/softmax.rs:23
Function
bench_matmul
(b: &mut Bencher)
benches/softmax.rs:55
Function
bench_softmax
(b: &mut Bencher)
benches/softmax.rs:11
Function
embedding
()
src/model/smelt.rs:531
Method
empty_past_key_values
(&self)
src/model/smelt.rs:471
Method
fmt
(&self, f: &mut std::fmt::Formatter<'_>)
src/model/smelt.rs:315
Method
from
(weight: TensorView<'a>, bias: TensorView<'a>)
src/model/smelt.rs:327
Method
from_tensors
( index: usize, tensors: &SafeTensors, dev: &Dev, )
src/model/dfdx.rs:233
Method
from_tensors
(index: usize, tensors: &'a SafeTensors<'a>)
src/model/smelt.rs:159
Function
health
()
src/main.rs:112
Function
inference
((State(state), payload): (State<AppState>, String))
src/main.rs:116
Function
layer_norm
()
src/model/smelt.rs:562
Function
main
()
build.rs:110
Function
main
()
src/main.rs:62
Function
main
()
docker/dummy.rs:1
Function
main
()
examples/download.rs:5
Function
main
()
examples/run.rs:3
Function
mlp
()
src/model/smelt.rs:746
Method
new
(past_sequence_length: usize, dev: &Dev)
src/model/dfdx.rs:66
Method
new
(num_heads: usize, past_sequence_length: usize, head_dim: usize)
src/model/smelt.rs:14
Function
simple_attention
()
src/model/smelt.rs:625
Function
simple_attention_ops
()
src/model/smelt.rs:889
Function
simple_attention_qk
()
src/model/smelt.rs:789
Function
simple_logits
()
src/lib.rs:131
Function
simplify
(data: &[f32])
src/lib.rs:124
Function
soft
(A)
benches/softmax.py:2
Function
suggest_setvars_cmd
(root: &str)
build.rs:102
Function
tensor_values
()
src/model/smelt.rs:511
Function
test_addmm
(benchmark)
benches/softmax.py:11
Function
test_matmul
(benchmark)
benches/softmax.py:19
Function
test_softmax
(benchmark)
benches/softmax.py:5