MCPcopy Create free account

hub / github.com/Narsil/fast_gpt2 / functions

Functions50 in github.com/Narsil/fast_gpt2

↓ 7 callersMethodforward
(&self, tensor: &mut OwnedTensor)
src/model/smelt.rs:177
↓ 6 callersFunctiondownload
( url: &str, filename: &str, max_files: usize, chunk_size: usize, )
src/download.rs:9
↓ 4 callersMethodempty_past_key_values
(&self, dev: &Dev)
src/model/dfdx.rs:464
↓ 4 callersFunctionlinear_from
( weight: TensorView, bias: TensorView, dev: &Dev, )
src/model/dfdx.rs:78
↓ 3 callersMethodforward
(&self, tensor: FTensor<HiddenShape>)
src/model/dfdx.rs:251
↓ 3 callersFunctionlayer_norm_from
( weight: TensorView, bias: TensorView, dev: &Dev, )
src/model/dfdx.rs:101
↓ 2 callersFunctionattention
( qkv: &T, qk: &mut TM, max: &mut [f32], past: &mut PastKeyValue, out: &mut OwnedTensor, )
src/model/smelt.rs:23
↓ 2 callersFunctionembedding_from
( weight: TensorView, dev: &Dev, )
src/model/dfdx.rs:112
↓ 2 callersFunctionsplit_qkv
( qkv: &T, past: &PastKeyValue, )
src/model/smelt.rs:80
↓ 1 callersFunctionattention_reshape
( qkv: &FTensor<QkvShape>, past_key: &FTensor<PastKeyShape>, past_value: &FTensor<PastValueShape>,
src/model/dfdx.rs:121
↓ 1 callersFunctiondownload_chunk
( client: reqwest::Client, url: String, filename: String, start: usize, stop: usize, )
src/download.rs:56
↓ 1 callersFunctionget_model
(filename: &str)
src/main.rs:32
↓ 1 callersFunctionget_tokenizer
(filename: &str)
src/main.rs:50
↓ 1 callersFunctionleak_buffer
(buffer: Mmap)
src/main.rs:26
↓ 1 callersFunctionrun
()
src/lib.rs:39
↓ 1 callersFunctionspecial_argmax
Argmax of the last dimension of tensor `x `.
src/model/smelt.rs:454
↓ 1 callersFunctionunbiased_linear_from
( weight: TensorView, dev: &Dev, )
src/model/dfdx.rs:92
Functionattention_data
()
src/model/smelt.rs:602
Functionbench_add
(b: &mut Bencher)
benches/softmax.rs:42
Functionbench_addmm
(b: &mut Bencher)
benches/softmax.rs:23
Functionbench_matmul
(b: &mut Bencher)
benches/softmax.rs:55
Functionbench_softmax
(b: &mut Bencher)
benches/softmax.rs:11
Functionembedding
()
src/model/smelt.rs:531
Methodempty_past_key_values
(&self)
src/model/smelt.rs:471
Methodfmt
(&self, f: &mut std::fmt::Formatter<'_>)
src/model/smelt.rs:315
Methodfrom
(weight: TensorView<'a>, bias: TensorView<'a>)
src/model/smelt.rs:327
Methodfrom_tensors
( index: usize, tensors: &SafeTensors, dev: &Dev, )
src/model/dfdx.rs:233
Methodfrom_tensors
(index: usize, tensors: &'a SafeTensors<'a>)
src/model/smelt.rs:159
Functionhealth
()
src/main.rs:112
Functioninference
((State(state), payload): (State<AppState>, String))
src/main.rs:116
Functionlayer_norm
()
src/model/smelt.rs:562
Functionmain
()
build.rs:110
Functionmain
()
src/main.rs:62
Functionmain
()
docker/dummy.rs:1
Functionmain
()
examples/download.rs:5
Functionmain
()
examples/run.rs:3
Functionmlp
()
src/model/smelt.rs:746
Methodnew
(past_sequence_length: usize, dev: &Dev)
src/model/dfdx.rs:66
Methodnew
(num_heads: usize, past_sequence_length: usize, head_dim: usize)
src/model/smelt.rs:14
Functionsimple_attention
()
src/model/smelt.rs:625
Functionsimple_attention_ops
()
src/model/smelt.rs:889
Functionsimple_attention_qk
()
src/model/smelt.rs:789
Functionsimple_logits
()
src/lib.rs:131
Functionsimplify
(data: &[f32])
src/lib.rs:124
Functionsoft
(A)
benches/softmax.py:2
Functionsuggest_setvars_cmd
(root: &str)
build.rs:102
Functiontensor_values
()
src/model/smelt.rs:511
Functiontest_addmm
(benchmark)
benches/softmax.py:11
Functiontest_matmul
(benchmark)
benches/softmax.py:19
Functiontest_softmax
(benchmark)
benches/softmax.py:5