Functions — github.com/Noeda/rllama

hub / github.com/Noeda/rllama / functions

Functions465 in github.com/Noeda/rllama

⨍ Functions 465◇ Types & classes 55

↓ 1 callersFunctioncompute_freqs_cis

(dim: usize, end: usize, theta: f64)

src/transformer.rs:858

↓ 1 callersMethoddata_u16_from_gpu

(&mut self, data: *mut u16)

src/tensor_opencl_support.rs:198

↓ 1 callersMethoddata_u16_to_gpu

( &self, data: *const u16, data_layout: Layout, nitems: usize, rows: i

src/tensor_opencl_support.rs:134

↓ 1 callersMethoddebug_print_go

(&self, indent: usize)

src/unpickler.rs:271

↓ 1 callersMethoddont_use_opencl

(mut self)

src/transformer.rs:84

↓ 1 callersFunctiongather_f32x8

(ptr: *const f32, indices: I32x8)

src/simd_support.rs:36

↓ 1 callersMethodget

(&self, tokens: &[TokenId])

src/rllama_main.rs:606

↓ 1 callersMethodget_embedding

(&self, idx: usize)

src/embedding.rs:50

↓ 1 callersMethodget_global

(&self)

src/unpickler.rs:110

↓ 1 callersMethodget_persistent_id

(&self)

src/unpickler.rs:131

↓ 1 callersMethodget_str_key2

Same as get_str_key but tries two keys, returning the first one that is found.

src/unpickler.rs:53

↓ 1 callersMethodhadamard_product_broadcast

(&self, other: &Tensor)

src/tensor.rs:654

↓ 1 callersMethodhadamard_product_gpu

(&self, other: &Tensor)

src/tensor.rs:701

↓ 1 callersMethodhadamard_product_inplace

( &mut self, other: &OpenCLTensor, )

src/tensor_opencl_support.rs:258

↓ 1 callersFunctioni32x8_from_values

( val0: i32, val1: i32, val2: i32, val3: i32, val4: i32, val5: i32, val6: i32,

src/simd_support.rs:70

↓ 1 callersMethodinput

(&self)

src/protomodels/sentencepiece_model.rs:2070

↓ 1 callersMethodkeys

Returns all keys as a set of strings, if the value is a dictionary. Otherwise returns empty set.

src/unpickler.rs:70

↓ 1 callersMethodlimit_size

Makes sure the cache repository is not larger than sz, evicts any older items.

src/rllama_main.rs:585

↓ 1 callersMethodlogits_to_btreemap

( &self, logits: &Tensor, tokenizer: &Tokenizer, )

src/token_sampler.rs:68

↓ 1 callersFunctionmain

()

examples/api_hello_world.py:11

↓ 1 callersFunctionmake_programs

(ctx: &Context, queue: &Queue)

src/tensor_opencl_support.rs:414

↓ 1 callersMethodmatrix_mul_inplace_transposed_gpu

(&mut self, src: &Tensor, other: &Tensor)

src/tensor.rs:1230

↓ 1 callersMethodmatrix_vector_mul_transposed_f16

(&self, other: &Tensor)

src/tensor.rs:1719

↓ 1 callersMethodmatrix_vector_mul_transposed_f32

(&self, other: &Tensor)

src/tensor.rs:1837

↓ 1 callersMethodmean_cols

Computes mean for each row, so that columns become 1.

src/tensor.rs:511

↓ 1 callersMethodneed_to_do_antitranspose

(&self)

src/data_source.rs:223

↓ 1 callersMethodnum_threads

(&self)

src/protomodels/sentencepiece_model.rs:480

↓ 1 callersMethodpercentage_to_gpu

(mut self, percentage: f32)

src/transformer.rs:91

↓ 1 callersMethodpow

(&self, power: f32)

src/tensor.rs:537

↓ 1 callersMethodrsqrt

(&self)

src/tensor.rs:561

↓ 1 callersMethodscalar_multiply_broadcast

(&self, other: &Tensor)

src/tensor.rs:619

↓ 1 callersFunctionserver_inference

( cli: Cli, tr: Arc<Transformer>, tok: Arc<Tokenizer>, be_quiet: bool, max_seq_len: usize,

src/rllama_main.rs:319

↓ 1 callersMethodsilu_gpu

(&self)

src/tensor.rs:806

↓ 1 callersMethodsilu_inplace

(&mut self)

src/tensor_opencl_support.rs:283

↓ 1 callersMethodstr_to_id

Tries to find a token from dictionary.

src/tokenizer.rs:109

↓ 1 callersMethodto_f16_naive

Naive implementation of to_f16, used for testing that the faster methods are correct.

src/tensor.rs:2264

↓ 1 callersMethodto_f32_naive

Naive implementation of to_f32, used for testing that the faster methods are correct.

src/tensor.rs:2209

↓ 1 callersMethodto_tensor_builder2

(&self, tensor_name: String, args: &[Value])

src/unpickler.rs:166

↓ 1 callersMethodto_vec

Converts the tensor to two-dimensional Vec<f32>. Meant for debugging and making it easy to print tensors.

src/tensor.rs:385

↓ 1 callersMethodtokenize_to_pieces

Converts a string to a Vec<&str> You may want to use tokenize_to_ids instead. This will not add start or end tokens; only the string is processed. I

src/tokenizer.rs:125

↓ 1 callersMethodtranspose_from

(&mut self, other: &OpenCLTensor)

src/tensor_opencl_support.rs:234

↓ 1 callersMethodtranspose_gpu

(&self)

src/tensor.rs:927

↓ 1 callersMethoduse_opencl

(mut self)

src/transformer.rs:74

Methodadd_dummy_prefix

(&self)

src/protomodels/sentencepiece_model.rs:1732

Methodallow_whitespace_only_pieces

(&self)

src/protomodels/sentencepiece_model.rs:613

Methodbos_id

(&self)

src/protomodels/sentencepiece_model.rs:782

Methodbos_piece

(&self)

src/protomodels/sentencepiece_model.rs:875

Methodbyte_fallback

(&self)

src/protomodels/sentencepiece_model.rs:687

Methodcharacter_coverage

(&self)

src/protomodels/sentencepiece_model.rs:328

Methodclear

(&mut self)

src/protomodels/sentencepiece_model.rs:1477

Methodclear_add_dummy_prefix

(&mut self)

src/protomodels/sentencepiece_model.rs:1736

Methodclear_allow_whitespace_only_pieces

(&mut self)

src/protomodels/sentencepiece_model.rs:617

Methodclear_bos_id

(&mut self)

src/protomodels/sentencepiece_model.rs:786

Methodclear_bos_piece

(&mut self)

src/protomodels/sentencepiece_model.rs:882

Methodclear_byte_fallback

(&mut self)

src/protomodels/sentencepiece_model.rs:691

Methodclear_character_coverage

(&mut self)

src/protomodels/sentencepiece_model.rs:332

Methodclear_differential_privacy_clipping_threshold

(&mut self)

src/protomodels/sentencepiece_model.rs:313

Methodclear_differential_privacy_noise_level

(&mut self)

src/protomodels/sentencepiece_model.rs:294

Methodclear_enable_differential_privacy

(&mut self)

src/protomodels/sentencepiece_model.rs:275

Methodclear_eos_id

(&mut self)

src/protomodels/sentencepiece_model.rs:805

Methodclear_eos_piece

(&mut self)

src/protomodels/sentencepiece_model.rs:918

Methodclear_escape_whitespaces

(&mut self)

src/protomodels/sentencepiece_model.rs:1774

Methodclear_expected

(&mut self)

src/protomodels/sentencepiece_model.rs:2113

Methodclear_hard_vocab_limit

(&mut self)

src/protomodels/sentencepiece_model.rs:729

Methodclear_input

(&mut self)

src/protomodels/sentencepiece_model.rs:2077

Methodclear_input_format

(&mut self)

src/protomodels/sentencepiece_model.rs:146

Methodclear_input_sentence_size

(&mut self)

src/protomodels/sentencepiece_model.rs:351

Methodclear_max_sentence_length

(&mut self)

src/protomodels/sentencepiece_model.rs:465

Methodclear_max_sentencepiece_length

(&mut self)

src/protomodels/sentencepiece_model.rs:522

Methodclear_mining_sentence_size

(&mut self)

src/protomodels/sentencepiece_model.rs:389

Methodclear_model_prefix

(&mut self)

src/protomodels/sentencepiece_model.rs:182

Methodclear_model_type

(&mut self)

src/protomodels/sentencepiece_model.rs:218

Methodclear_name

(&mut self)

src/protomodels/sentencepiece_model.rs:1667

Methodclear_normalization_rule_tsv

(&mut self)

src/protomodels/sentencepiece_model.rs:1796

Methodclear_num_sub_iterations

(&mut self)

src/protomodels/sentencepiece_model.rs:503

Methodclear_num_threads

(&mut self)

src/protomodels/sentencepiece_model.rs:484

Methodclear_pad_id

(&mut self)

src/protomodels/sentencepiece_model.rs:824

Methodclear_pad_piece

(&mut self)

src/protomodels/sentencepiece_model.rs:954

Methodclear_piece

(&mut self)

src/protomodels/sentencepiece_model.rs:2429

Methodclear_precompiled_charsmap

(&mut self)

src/protomodels/sentencepiece_model.rs:1703

Methodclear_remove_extra_whitespaces

(&mut self)

src/protomodels/sentencepiece_model.rs:1755

Methodclear_required_chars

(&mut self)

src/protomodels/sentencepiece_model.rs:658

Methodclear_score

(&mut self)

src/protomodels/sentencepiece_model.rs:2462

Methodclear_seed_sentencepiece_size

(&mut self)

src/protomodels/sentencepiece_model.rs:427

Methodclear_self_test_sample_size

(&mut self)

src/protomodels/sentencepiece_model.rs:256

Methodclear_shrinking_factor

(&mut self)

src/protomodels/sentencepiece_model.rs:446

Methodclear_shuffle_input_sentence

(&mut self)

src/protomodels/sentencepiece_model.rs:370

Methodclear_split_by_number

(&mut self)

src/protomodels/sentencepiece_model.rs:560

Methodclear_split_by_unicode_script

(&mut self)

src/protomodels/sentencepiece_model.rs:541

Methodclear_split_by_whitespace

(&mut self)

src/protomodels/sentencepiece_model.rs:579

Methodclear_split_digits

(&mut self)

src/protomodels/sentencepiece_model.rs:636

Methodclear_train_extremely_large_corpus

(&mut self)

src/protomodels/sentencepiece_model.rs:1023

Methodclear_training_sentence_size

(&mut self)

src/protomodels/sentencepiece_model.rs:408

Methodclear_treat_whitespace_as_suffix

(&mut self)

src/protomodels/sentencepiece_model.rs:598

Methodclear_type_

(&mut self)

src/protomodels/sentencepiece_model.rs:2484

Methodclear_unk_id

(&mut self)

src/protomodels/sentencepiece_model.rs:767

Methodclear_unk_piece

(&mut self)

src/protomodels/sentencepiece_model.rs:846

Methodclear_unk_surface

(&mut self)

src/protomodels/sentencepiece_model.rs:990

Methodclear_use_all_vocab

(&mut self)

src/protomodels/sentencepiece_model.rs:748

Methodclear_vocab_size

(&mut self)

src/protomodels/sentencepiece_model.rs:237

← previous next →101–200 of 465, ranked by callers