MCPcopy Create free account

hub / github.com/crabml/crabml / functions

Functions598 in github.com/crabml/crabml

⨍ Functions 598◇ Types & classes 96
↓ 1 callersFunctionvec_dot_f16_f16_fallback
( a: &[f16], a_offset: usize, b: &[f16], b_offset: usize, len: usize, )
crabml-core/src/cpu/buf/buf_f16.rs:83
↓ 1 callersFunctionvec_dot_f16_f16_simd
( a: &[f16], a_offset: usize, b: &[f16], b_offset: usize, k: usize, )
crabml-core/src/cpu/buf/buf_f16.rs:54
↓ 1 callersFunctionvec_dot_f16_f16_strided_fallback
( a: &[f16], a_base: usize, a_stride: usize, k: usize, b: &[f16], )
crabml-core/src/cpu/buf/buf_f16.rs:242
↓ 1 callersFunctionvec_dot_f16_f16_strided_simd
( a: &[f16], a_base: usize, a_stride: usize, k: usize, b: &[f16], )
crabml-core/src/cpu/buf/buf_f16.rs:193
↓ 1 callersFunctionvec_dot_f32_f32
(a: &[f32], a_offset: usize, b: &[f32], b_offset: usize, len: usize)
crabml-core/src/cpu/buf/buf_f32.rs:19
↓ 1 callersFunctionvec_dot_f32_f32_strided_fallback
( a: &[f32], a_base: usize, a_stride: usize, k: usize, b: &[f32], )
crabml-core/src/cpu/buf/buf_f32.rs:69
↓ 1 callersFunctionvec_dot_q4_0_q8_0
(abs: &[BlockQ4_0], bbs: &[BlockQ8_0])
crabml-core/src/cpu/buf/buf_q4_0.rs:126
↓ 1 callersFunctionvec_dot_q4_0_q8_0_avx2
(abs: &[BlockQ4_0], bbs: &[BlockQ8_0])
crabml-core/src/cpu/buf/buf_q4_0.rs:215
↓ 1 callersFunctionvec_dot_q4_0_q8_0_neon
(abs: &[BlockQ4_0], bbs: &[BlockQ8_0])
crabml-core/src/cpu/buf/buf_q4_0.rs:145
↓ 1 callersFunctionvec_dot_q4_1_q8_1
(abs: &[BlockQ4_1], bbs: &[BlockQ8_1])
crabml-core/src/cpu/buf/buf_q4_1.rs:126
↓ 1 callersFunctionvec_dot_q4_1_q8_1_avx2
(abs: &[BlockQ4_1], bbs: &[BlockQ8_1])
crabml-core/src/cpu/buf/buf_q4_1.rs:214
↓ 1 callersFunctionvec_dot_q4_1_q8_1_neon
(abs: &[BlockQ4_1], bbs: &[BlockQ8_1])
crabml-core/src/cpu/buf/buf_q4_1.rs:145
↓ 1 callersFunctionvec_dot_q5_0_q8_0
(abs: &[BlockQ5_0], bbs: &[BlockQ8_0])
crabml-core/src/cpu/buf/buf_q5_0.rs:145
↓ 1 callersFunctionvec_dot_q5_0_q8_0_avx2
(abs: &[BlockQ5_0], bbs: &[BlockQ8_0])
crabml-core/src/cpu/buf/buf_q5_0.rs:219
↓ 1 callersFunctionvec_dot_q5_0_q8_0_fallback
(abs: &[BlockQ5_0], bbs: &[BlockQ8_0])
crabml-core/src/cpu/buf/buf_q5_0.rs:274
↓ 1 callersFunctionvec_dot_q5_0_q8_0_neon
(abs: &[BlockQ5_0], bbs: &[BlockQ8_0])
crabml-core/src/cpu/buf/buf_q5_0.rs:164
↓ 1 callersFunctionvec_dot_q5_1_q8_1
(abs: &[BlockQ5_1], bbs: &[BlockQ8_1])
crabml-core/src/cpu/buf/buf_q5_1.rs:142
↓ 1 callersFunctionvec_dot_q5_1_q8_1_avx2
(abs: &[BlockQ5_1], bbs: &[BlockQ8_1])
crabml-core/src/cpu/buf/buf_q5_1.rs:212
↓ 1 callersFunctionvec_dot_q5_1_q8_1_fallback
(abs: &[BlockQ5_1], bbs: &[BlockQ8_1])
crabml-core/src/cpu/buf/buf_q5_1.rs:265
↓ 1 callersFunctionvec_dot_q5_1_q8_1_neon
(abs: &[BlockQ5_1], bbs: &[BlockQ8_1])
crabml-core/src/cpu/buf/buf_q5_1.rs:161
↓ 1 callersFunctionvec_dot_q6_k_q8_k
(abs: &[BlockQ6K], bbs: &[BlockQ8K])
crabml-core/src/cpu/buf/buf_q6_k.rs:183
↓ 1 callersFunctionvec_dot_q8_0_q8_0_avx2
(abs: &[BlockQ8_0], bbs: &[BlockQ8_0])
crabml-core/src/cpu/buf/buf_q8_0.rs:228
↓ 1 callersFunctionvec_dot_q8_0_q8_0_fallback
(abs: &[BlockQ8_0], bbs: &[BlockQ8_0])
crabml-core/src/cpu/buf/buf_q8_0.rs:275
↓ 1 callersFunctionvec_dot_q8_0_q8_0_neon
(abs: &[BlockQ8_0], bbs: &[BlockQ8_0])
crabml-core/src/cpu/buf/buf_q8_0.rs:155
↓ 1 callersFunctionvec_dot_q8_k_q8_k
(abs: &[BlockQ8K], bbs: &[BlockQ8K])
crabml-core/src/cpu/buf/buf_q8_k.rs:133
↓ 1 callersFunctionvec_dot_q8_k_q8_k_avx2
(abs: &[BlockQ8K], bbs: &[BlockQ8K])
crabml-core/src/cpu/buf/buf_q8_k.rs:179
↓ 1 callersFunctionvec_dot_q8_k_q8_k_fallback
(abs: &[BlockQ8K], bbs: &[BlockQ8K])
crabml-core/src/cpu/buf/buf_q8_k.rs:211
↓ 1 callersFunctionvec_dot_q8_k_q8_k_neon
(abs: &[BlockQ8K], bbs: &[BlockQ8K])
crabml-core/src/cpu/buf/buf_q8_k.rs:152
↓ 1 callersMethodvec_dot_rhs_dtype
(&self)
crabml-core/src/cpu/buf/api.rs:142
↓ 1 callersFunctionvec_fma_f16_f16_fallback
(a: &[f16], b: f16, c: &mut [f16], a_offset: usize, m: usize)
crabml-core/src/cpu/buf/buf_f16.rs:152
↓ 1 callersFunctionvec_fma_f16_f16_neon
(a: &[f16], b: f16, c: &mut [f16], a_offset: usize, m: usize)
crabml-core/src/cpu/buf/buf_f16.rs:130
↓ 1 callersFunctionvget_high_f16_f32
(i: float16x8_t)
crabml-core/src/cpu/archutil/aarch64.rs:48
↓ 1 callersFunctionvget_low_f16_f32
(i: float16x8_t)
crabml-core/src/cpu/archutil/aarch64.rs:62
↓ 1 callersMethodvocab
(&self)
crabml-core/src/tokenizer/mod.rs:80
↓ 1 callersMethodweights
(&self)
crabml-llama2/src/model.rs:122
↓ 1 callersMethodwith_debug_named_tensors
(mut self, debug_named_tensors: bool)
crabml-core/src/cpu/cpu_device.rs:39
↓ 1 callersMethodwith_device_options
(mut self, options: CpuTensorDeviceOptions)
crabml-llama2/src/model.rs:178
↓ 1 callersMethodwith_probability
(mut self, probability: f32)
crabml-llama2/src/model.rs:168
↓ 1 callersMethodwith_temperature
(mut self, temperature: f32)
crabml-llama2/src/model.rs:163
↓ 1 callersMethodwith_thread_num
(mut self, thread_num: usize)
crabml-core/src/cpu/cpu_device.rs:34
Methodadd_inplace
(mut self, b: &Self)
crabml-core/src/cpu/cpu_tensor.rs:396
Methodalloc
(shape: &[usize], dtype: GGMLType, device: Self::DeviceRef)
crabml-vulkan/src/vulkan_tensor.rs:71
Methodalloc
(shape: &[usize], dtype: GGMLType, device: Self::DeviceRef)
crabml-wgpu/src/wgpu_tensor.rs:89
Methodalloc
(shape: &[usize], dtype: GGMLType, device: Self::DeviceRef)
crabml-core/src/cpu/cpu_tensor.rs:138
Methodas_bytes
(&self)
crabml-core/src/cpu/buf/buf_q8_0.rs:52
Methodas_bytes
(&self)
crabml-core/src/cpu/buf/buf_q5_0.rs:65
Methodas_bytes
(&self)
crabml-core/src/cpu/buf/buf_q8_1.rs:41
Methodas_bytes
(&self)
crabml-core/src/cpu/buf/buf_q2_k.rs:103
Methodas_bytes
(&self)
crabml-core/src/cpu/buf/buf_q8_k.rs:49
Methodas_bytes
(&self)
crabml-core/src/cpu/buf/buf_q5_k.rs:88
Methodas_bytes
(&self)
crabml-core/src/cpu/buf/buf_q5_1.rs:68
Methodas_bytes
(&self)
crabml-core/src/cpu/buf/buf_q4_0.rs:55
Methodas_bytes
(&self)
crabml-core/src/cpu/buf/buf_q3_k.rs:118
Methodas_bytes
(&self)
crabml-core/src/cpu/buf/buf_q6_k.rs:76
Methodas_bytes
(&self)
crabml-core/src/cpu/buf/buf_q4_1.rs:59
Methodas_bytes
(&self)
crabml-core/src/cpu/buf/buf_q4_k.rs:76
Methodat
(&self, idx: &[usize])
crabml-core/src/tensor/strider.rs:65
Methodbatch_matmul
(b, m, k) @ (b, k, n) -> (b, m, n)
crabml-core/src/cpu/cpu_tensor.rs:352
Functionbinary_inplace
( buf1: &mut CpuTensorBuf<'a>, buf2: &CpuTensorBuf<'a>, strider1: &TensorStrider, strider2: &T
crabml-core/src/cpu/primitives/arithmetic.rs:81
Methodcompute_strides
(shape: &[usize])
crabml-core/src/tensor/strider.rs:216
Methodconcatenate
(&mut self, rhs: &Self, axis: usize)
crabml-core/src/cpu/cpu_tensor.rs:251
Methodconf
(&self)
crabml-llama2/src/model.rs:114
Methodconvert_cpu_tensor
(tensor: &CpuTensor, device: T::DeviceRef)
crabml-llama2/src/model.rs:817
Methodconvert_cpu_weights
( weights: &LlamaWeights<CpuTensor>, device: T::DeviceRef, )
crabml-llama2/src/model.rs:687
Methodconvert_tensor_infos
( tensor_infos: &[GGUFOnDiskTensorInfo], tensor_data: &'a [u8], )
crabml-core/src/gguf.rs:737
Methoddecode
(buf: &mut GGUFBufReader<'a>)
crabml-core/src/gguf.rs:522
Methoddecode
(&self, token: TokenID)
crabml-core/src/tokenizer/tokenizer_llama.rs:38
Methoddefault
()
crabml-vulkan/src/vulkan_device.rs:71
Methoddefault
()
crabml-wgpu/src/wgpu_device.rs:16
Methoddefault
()
crabml-core/src/cpu/cpu_device.rs:24
Methoddefault
()
crabml-core/src/cpu/buf/buf_q2_k.rs:73
Methoddefault
()
crabml-core/src/cpu/buf/buf_q3_k.rs:88
Methoddefault
()
crabml-core/src/tokenizer/mod.rs:119
Methoddefault
()
crabml-llama2/src/model.rs:148
Methoddequantize
(self, dtype: GGMLType)
crabml-core/src/cpu/cpu_tensor.rs:64
Methoddequantize
(&self, buf: &mut [f32])
crabml-core/src/cpu/buf/buf_q8_0.rs:18
Methoddequantize
(&'a self, start: usize)
crabml-core/src/cpu/buf/buf_q8_1.rs:57
Methoddequantize
(&self, buf: &mut [f32])
crabml-core/src/cpu/buf/buf_q4_0.rs:18
Functiondiv_inplace
( buf1: &mut CpuTensorBuf<'a>, buf2: &CpuTensorBuf<'a>, strider1: &TensorStrider, strider2: &T
crabml-core/src/cpu/primitives/arithmetic.rs:71
Methoddrop
(&mut self)
crabml-core/src/tensor/metrics.rs:188
Methoddtype
(&self)
crabml-vulkan/src/vulkan_tensor.rs:120
Methoddump_debug_tensor
(&self, name: &str)
crabml-vulkan/src/vulkan_device.rs:194
Methoddump_debug_tensor
(&self, name: &str)
crabml-core/src/cpu/cpu_device.rs:96
Methoddup
(&self)
crabml-core/src/cpu/cpu_tensor.rs:333
Methodexport
(&self, dst: &mut [f32])
crabml-core/src/cpu/cpu_tensor.rs:339
Methodfmt
(&self, f: &mut std::fmt::Formatter<'_>)
crabml-cli/src/main.rs:73
Methodfmt
(&self, f: &mut std::fmt::Formatter)
crabml-core/src/gguf.rs:75
Methodfmt
(&self, f: &mut fmt::Formatter)
crabml-core/src/error.rs:43
Methodfrom
(buf: Vec<f32>)
crabml-core/src/cpu/buf/api.rs:404
Methodfrom_bytes
( buf: &'a [u8], typ: GGMLType, shape: &[usize], device: CpuTensorDeviceRef<'a
crabml-core/src/cpu/cpu_tensor.rs:48
Methodfrom_bytes
(data: &'a [u8])
crabml-core/src/cpu/buf/buf_q8_0.rs:32
Methodfrom_bytes
(data: &'a [u8])
crabml-core/src/cpu/buf/buf_q5_0.rs:45
Methodfrom_bytes
(data: &'a [u8])
crabml-core/src/cpu/buf/buf_q8_1.rs:21
Methodfrom_bytes
(data: &'a [u8])
crabml-core/src/cpu/buf/buf_q2_k.rs:89
Methodfrom_bytes
(data: &'a [u8])
crabml-core/src/cpu/buf/buf_q8_k.rs:29
Methodfrom_bytes
(data: &'a [u8])
crabml-core/src/cpu/buf/buf_q5_k.rs:69
Methodfrom_bytes
(data: &'a [u8])
crabml-core/src/cpu/buf/buf_q5_1.rs:44
Methodfrom_bytes
(data: &'a [u8])
crabml-core/src/cpu/buf/buf_q4_0.rs:36
Methodfrom_bytes
(data: &'a [u8])
crabml-core/src/cpu/buf/buf_q3_k.rs:104
Methodfrom_bytes
(data: &'a [u8])
crabml-core/src/cpu/buf/buf_q6_k.rs:56
← previousnext →301–400 of 598, ranked by callers