MCPcopy Create free account

hub / github.com/crabml/crabml / functions

Functions598 in github.com/crabml/crabml

Functiontest_generate_q8_0
()
crabml-llama2/src/chat.rs:308
Functiontest_generate_q8_0
()
crabml-llama2/src/llama2.rs:674
Functiontest_generate_q8_0_with_f16_kvcache
()
crabml-llama2/src/llama2.rs:706
Functiontest_get_scale_min_k4
()
crabml-core/src/cpu/buf/util.rs:352
Functiontest_gguf_tokenizer
()
crabml-core/src/tokenizer/tokenizer_llama.rs:146
Functiontest_gpt2_tokenizer
()
crabml-core/src/tokenizer/tokenizer_gpt2.rs:227
Functiontest_is_contiguous
()
crabml-core/src/tensor/strider.rs:301
Functiontest_load_metadata
()
crabml-core/src/gguf.rs:917
Functiontest_load_q8_0
()
crabml-llama2/src/model.rs:850
Functiontest_load_tensors
()
crabml-core/src/gguf.rs:840
Functiontest_matmul
()
crabml-core/src/cpu/cpu_tensor.rs:530
Functiontest_matmul_vec
()
crabml-vulkan/src/vulkan_tensor.rs:767
Functiontest_nearest_i32
()
crabml-core/src/cpu/buf/util.rs:328
Functiontest_q2_k_quantize
()
crabml-core/src/cpu/buf/buf_q2_k.rs:269
Functiontest_q2_k_vec_dot_q8_k
()
crabml-core/src/cpu/buf/buf_q2_k.rs:281
Functiontest_q3_k_quantize
()
crabml-core/src/cpu/buf/buf_q3_k.rs:340
Functiontest_q3_k_vec_dot_q8_k
()
crabml-core/src/cpu/buf/buf_q3_k.rs:352
Functiontest_q4_0_block
()
crabml-core/src/cpu/buf/buf_q4_0.rs:260
Functiontest_q4_1_block
()
crabml-core/src/cpu/buf/buf_q4_1.rs:287
Functiontest_q4_1_quantize
()
crabml-core/src/cpu/buf/buf_q4_1.rs:316
Functiontest_q4_k_quantize
()
crabml-core/src/cpu/buf/buf_q4_k.rs:291
Functiontest_q4_k_vec_dot_q8_k
()
crabml-core/src/cpu/buf/buf_q4_k.rs:303
Functiontest_q5_0_block
()
crabml-core/src/cpu/buf/buf_q5_0.rs:306
Functiontest_q5_0_quantize
()
crabml-core/src/cpu/buf/buf_q5_0.rs:332
Functiontest_q5_1_block
()
crabml-core/src/cpu/buf/buf_q5_1.rs:298
Functiontest_q5_1_quantize
()
crabml-core/src/cpu/buf/buf_q5_1.rs:328
Functiontest_q5_k_quantize
()
crabml-core/src/cpu/buf/buf_q5_k.rs:332
Functiontest_q6_k_block
()
crabml-core/src/cpu/buf/buf_q6_k.rs:247
Functiontest_q6_k_quantize
()
crabml-core/src/cpu/buf/buf_q6_k.rs:279
Functiontest_q6_k_quantize_2
()
crabml-core/src/cpu/buf/buf_q6_k.rs:291
Functiontest_q80_block
()
crabml-core/src/cpu/buf/buf_q8_0.rs:293
Functiontest_q8_1_block
()
crabml-core/src/cpu/buf/buf_q8_1.rs:136
Functiontest_q8_k_block
()
crabml-core/src/cpu/buf/buf_q8_k.rs:233
Functiontest_q8_k_quantize
()
crabml-core/src/cpu/buf/buf_q8_k.rs:265
Functiontest_resize
()
crabml-core/src/cpu/cpu_tensor.rs:603
Functiontest_rms_norm
()
crabml-core/src/cpu/cpu_tensor.rs:484
Functiontest_rope
()
crabml-vulkan/src/vulkan_tensor.rs:643
Functiontest_rope
()
crabml-core/src/cpu/cpu_tensor.rs:509
Functiontest_scale_inplace
()
crabml-vulkan/src/vulkan_tensor.rs:536
Functiontest_silu
()
crabml-core/src/cpu/cpu_tensor.rs:558
Functiontest_silu_inplace
()
crabml-vulkan/src/vulkan_tensor.rs:556
Functiontest_softmax
()
crabml-vulkan/src/vulkan_tensor.rs:591
Functiontest_softmax
()
crabml-core/src/cpu/cpu_tensor.rs:544
Functiontest_split_words
()
crabml-core/src/tokenizer/tokenizer_gpt2.rs:274
Functiontest_strider_increment_pos
()
crabml-core/src/tensor/strider.rs:310
Functiontest_strider_iter_axis
()
crabml-core/src/tensor/strider.rs:260
Functiontest_strider_reshape
()
crabml-core/src/tensor/strider.rs:243
Functiontest_strider_resize
()
crabml-core/src/tensor/strider.rs:326
Functiontest_tensor_rms_norm
()
crabml-vulkan/src/vulkan_tensor.rs:616
Functiontest_tensor_view
()
crabml-core/src/cpu/cpu_tensor.rs:456
Functiontest_transpose
()
crabml-core/src/tensor/strider.rs:279
Functiontest_unicode_table
()
crabml-core/src/tokenizer/tokenizer_gpt2.rs:290
Functiontest_vec_dot_q5_k_q8_k
()
crabml-core/src/cpu/buf/buf_q5_k.rs:342
Functiontest_vec_dot_q8_0_q8_0
()
crabml-core/src/cpu/buf/buf_q8_0.rs:325
Functiontest_vec_fma_f16_f16
()
crabml-core/src/cpu/buf/buf_f16.rs:270
Functiontest_wgpu_batch_matmul
()
crabml-wgpu/src/wgpu_tensor.rs:897
Functiontest_wgpu_concatenate
()
crabml-wgpu/src/wgpu_tensor.rs:940
Functiontest_wgpu_concatenate2
()
crabml-wgpu/src/wgpu_tensor.rs:971
Functiontest_wgpu_copy_from
()
crabml-wgpu/src/wgpu_tensor.rs:834
Functiontest_wgpu_gelu
()
crabml-wgpu/src/wgpu_tensor.rs:1039
Functiontest_wgpu_matmul
()
crabml-wgpu/src/wgpu_tensor.rs:880
Functiontest_wgpu_rope
()
crabml-wgpu/src/wgpu_tensor.rs:918
Functiontest_wgpu_silu
()
crabml-wgpu/src/wgpu_tensor.rs:1021
Functiontest_wgpu_softmax
()
crabml-wgpu/src/wgpu_tensor.rs:1003
Functiontest_wgpu_tensor_add
()
crabml-wgpu/src/wgpu_tensor.rs:775
Functiontest_wgpu_tensor_alloc
()
crabml-wgpu/src/wgpu_tensor.rs:809
Functiontest_wgpu_tensor_mul
()
crabml-wgpu/src/wgpu_tensor.rs:788
Functiontest_wgpu_tensor_new_and_export
()
crabml-wgpu/src/wgpu_tensor.rs:763
Functiontest_wgpu_tensor_rms_norm
()
crabml-wgpu/src/wgpu_tensor.rs:852
Functiontest_wgpu_tensor_with_name
()
crabml-wgpu/src/wgpu_tensor.rs:822
Methodtry_from
(v: u32)
crabml-core/src/gguf.rs:138
Methodtyp
(&self)
crabml-core/src/cpu/cpu_tensor.rs:78
Functionvaddq_f16
(a: float16x8_t, b: float16x8_t)
crabml-core/src/cpu/archutil/aarch64.rs:75
Functionvcvt_f16_f32
(i: float16x4_t)
crabml-core/src/cpu/archutil/aarch64.rs:22
Methodvec_dot
(&self, a_offset: usize, b: &Self, b_offset: usize, len: usize)
crabml-core/src/cpu/buf/buf_q8_0.rs:79
Methodvec_dot
(&self, a_offset: usize, b: &QuantBufQ8_0, b_offset: usize, len: usize)
crabml-core/src/cpu/buf/buf_q5_0.rs:92
Methodvec_dot
(&self, _a_offset: usize, _b: &Self, _b_offset: usize, _len: usize)
crabml-core/src/cpu/buf/buf_q8_1.rs:68
Methodvec_dot
(&self, a_offset: usize, b: &QuantBufQ8K, b_offset: usize, len: usize)
crabml-core/src/cpu/buf/buf_q2_k.rs:136
Methodvec_dot
(&self, a_offset: usize, b: &QuantBufQ8K, b_offset: usize, len: usize)
crabml-core/src/cpu/buf/buf_q8_k.rs:76
Methodvec_dot
(&self, a_offset: usize, b: &QuantBufQ8K, b_offset: usize, len: usize)
crabml-core/src/cpu/buf/buf_q5_k.rs:115
Methodvec_dot
(&self, a_offset: usize, b: &QuantBufQ8_1, b_offset: usize, len: usize)
crabml-core/src/cpu/buf/buf_q5_1.rs:91
Methodvec_dot
(&self, a_offset: usize, b: &QuantBufQ8_0, b_offset: usize, len: usize)
crabml-core/src/cpu/buf/buf_q4_0.rs:82
Methodvec_dot
(&self, a_offset: usize, b: &QuantBufQ8K, b_offset: usize, len: usize)
crabml-core/src/cpu/buf/buf_q3_k.rs:151
Methodvec_dot
(&self, a_offset: usize, b: &QuantBufQ8K, b_offset: usize, len: usize)
crabml-core/src/cpu/buf/buf_q6_k.rs:103
Methodvec_dot
(&self, a_offset: usize, b: &QuantBufQ8_1, b_offset: usize, len: usize)
crabml-core/src/cpu/buf/buf_q4_1.rs:86
Methodvec_dot
(&self, a_offset: usize, b: &QuantBufQ8K, b_offset: usize, len: usize)
crabml-core/src/cpu/buf/buf_q4_k.rs:103
Functionvec_dot_f16_f16_strided
( a: &[f16], a_base: usize, a_stride: usize, k: usize, b: &[f16], )
crabml-core/src/cpu/buf/buf_f16.rs:99
Functionvec_dot_f32_f32_strided
vec_dot_f32_f32_strided is called in batch_matmul_vec, which is used in the computation of the scaled dot product attention in the transformer model.
crabml-core/src/cpu/buf/buf_f32.rs:40
Methodversion
(&self)
crabml-core/src/gguf.rs:769
Functionvmulq_f16
(a: float16x8_t, b: float16x8_t)
crabml-core/src/cpu/archutil/aarch64.rs:89
Methodwith_debug_named_tensor
(mut self, v: bool)
crabml-vulkan/src/vulkan_device.rs:64
Methodwith_metrics
(mut self, metrics: TensorMetrics)
crabml-core/src/cpu/cpu_device.rs:44
Methodwith_name
(mut self, name: String)
crabml-vulkan/src/vulkan_tensor.rs:129
Methodwith_name
(mut self, name: String)
crabml-core/src/cpu/cpu_tensor.rs:232
Methodwith_options
(opts: CpuTensorDeviceOptions)
crabml-core/src/cpu/cpu_device.rs:69
Methodwith_staging_buf_bytes
(mut self, v: usize)
crabml-vulkan/src/vulkan_device.rs:59
Methodwith_strider
(self, strider: TensorStrider)
crabml-core/src/cpu/cpu_tensor.rs:223
Methodwith_thread_num
(mut self, thread_num: usize)
crabml-llama2/src/model.rs:173
← previous501–598 of 598, ranked by callers