Functions — github.com/Narsil/smelte-rs

CodeHub MCPcopy Create free account

hub / github.com/Narsil/smelte-rs / functions

Functions212 in github.com/Narsil/smelte-rs

⨍ Functions 212◇ Types & classes 58

↓ 1 callersFunctionunsplit

(ctx: &mut Gpt2Context<F32CudaTensor>)

src/nn/models/gpt2.rs:359

tensor elementwise addition. b += a. a is automatically broadcasted.

src/cpu/f16/ops.rs:180

(x: &Self, y: &mut Self)

src/gpu/f32/traits.rs:35

(x: &Self, y: &mut Self)

src/cpu/f32/traits.rs:34

Applies `func` to every item of the tensor

src/cpu/f16/ops.rs:371

Methodattention

( query: &Linear<F32Tensor>, key: &Linear<F32Tensor>, value: &Linear<F32Te

src/nn/models/bert.rs:152

Methodattention

( qkv: &LinearT<F32Tensor>, ctx: &mut Gpt2Context<F32Tensor>, i: usize,

src/nn/models/gpt2.rs:301

Functionbench_gelu

(b: &mut Bencher)

benches/ops.rs:8

Functionbench_normalize

(b: &mut Bencher)

benches/ops.rs:17

Functionbench_softmax

(b: &mut Bencher)

benches/ops.rs:26

src/nn/layers/linear.rs:30

Creates a new borrowed tensor with given shape. Can fail if data doesn't match the shape ``` use smelte_rs::cpu::f32::Tensor; let tensor = Tensor::bo

src/cpu/f32/tensor.rs:80

Creates a new borrowed tensor with given shape. Can fail if data doesn't match the shape ``` use smelte-rs::cpu::f16::Tensor; let data = [1.0, 2.0, 3

src/cpu/f16/tensor.rs:66

Methodbroadcast_add

(x: &Self, y: &mut Self)

src/gpu/f32/traits.rs:38

Methodbroadcast_add

(x: &Self, y: &mut Self)

src/cpu/f32/traits.rs:37

Methodbroadcast_mul

(x: &Self, y: &mut Self)

src/gpu/f32/traits.rs:47

Methodbroadcast_mul

(x: &Self, y: &mut Self)

src/cpu/f32/traits.rs:46

(src: &Self, dst: &mut Self)

src/gpu/f32/traits.rs:29

(src: &Self, dst: &mut Self)

src/cpu/f32/traits.rs:28

A slice to the underlying tensor data. Exists uniquely for symetry with gpu Tensor. ``` use smelte_rs::cpu::f32::Tensor; let tensor = Tensor::zeros(v

src/cpu/f32/tensor.rs:46

src/nn/models/bert.rs:164

src/nn/models/gpt2.rs:312

src/gpu/f32/traits.rs:16

src/cpu/f32/traits.rs:15

Functionfaster_gelu

src/cpu/f32/ops.rs:418

Functionfaster_gelu

src/cpu/f16/ops.rs:345

(cublas: CublasError)

src/gpu/f32/ops.rs:29

Creates a tensor from a cpu [Vec].

src/gpu/f32/tensor.rs:98

Creates a new tensor with given shape. Can fail if data doesn't match the shape Exists only for symetry with gpu tensor. ``` use smelte_rs::cpu::f32::

src/cpu/f32/tensor.rs:115

Methodfrom_tensors

(tensors: &'a SafeTensors<'a>, device: &Device)

examples/bert.rs:125

Methodfrom_tensors

(tensors: &'a SafeTensors<'a>, device: &Device)

examples/gpt2.rs:181

Functiong_matmul

( a: &Tensor, b: &Tensor, c: &mut Tensor, )

src/gpu/f32/ops.rs:97

Functiong_matmul

( a: &Tensor, b: &Tensor, c: &mut Tensor, )

src/cpu/f32/ops.rs:58

Functiong_matmul

( a: &Tensor, b: &Tensor, c: &mut Tensor, )

src/cpu/f16/ops.rs:70

Functiong_softmax

( x: &mut Tensor, past_sequence_length: usize, )

src/gpu/f32/ops.rs:377

Functiong_softmax

( x: &mut Tensor, past_sequence_length: usize, )

src/cpu/f32/ops.rs:313

Functiong_softmax

( x: &mut Tensor, past_sequence_length: usize, )

src/cpu/f16/ops.rs:250

src/cpu/f32/ops.rs:426

src/cpu/f16/ops.rs:359

(x: &mut Tensor)

src/gpu/f32/traits.rs:77

(x: &mut Tensor)

src/cpu/f32/traits.rs:76

examples/bert.rs:409

examples/gpt2.rs:285

(x: &Self, y: &Self, out: &mut Self)

src/gpu/f32/traits.rs:59

(x: &Self, y: &Self, out: &mut Self)

src/cpu/f32/traits.rs:58

(x: &Self, y: &Self, out: &mut Self)

src/gpu/f32/traits.rs:65

(x: &Self, y: &Self, out: &mut Self)

src/cpu/f32/traits.rs:64

tensor elementwise multiplication. b *= a. a is automatically broadcasted.

src/cpu/f16/ops.rs:206

(x: &Self, y: &mut Self)

src/gpu/f32/traits.rs:44

(x: &Self, y: &mut Self)

src/cpu/f32/traits.rs:43

src/gpu/f32/tensor.rs:24

Creates a new tensor with given shape. Can fail if data doesn't match the shape ``` use smelte_rs::cpu::f32::Tensor; let data = vec![1.0, 2.0, 3.0, 4

src/cpu/f32/tensor.rs:92

Creates a new tensor with given shape. Can fail if data doesn't match the shape ``` use smelte-rs::cpu::f16::Tensor; let data = vec![1.0, 2.0, 3.0, 4

src/cpu/f16/tensor.rs:78

Linear layer creation

src/nn/layers/linear.rs:13

src/nn/layers/embedding.rs:12

src/nn/layers/layer_norm.rs:14

src/nn/models/bert.rs:340

src/nn/models/gpt2.rs:48

Methodnormalize

(x: &mut Self, epsilon: f32)

src/gpu/f32/traits.rs:53

Methodnormalize

(x: &mut Self, epsilon: f32)

src/cpu/f32/traits.rs:52

src/nn/models/bert.rs:64

src/nn/models/gpt2.rs:183

src/nn/models/gpt2.rs:644

(x: &[usize], weight: &Self, out: &mut Self)

src/gpu/f32/traits.rs:71

(x: &[usize], weight: &Self, out: &mut Self)

src/cpu/f32/traits.rs:70

Methodset_num_heads

src/nn/models/gpt2.rs:590

src/gpu/f32/traits.rs:12

src/cpu/f32/traits.rs:11

Functionsimple_add

src/gpu/f32/ops.rs:568

Functionsimple_broadcast_add

src/gpu/f32/ops.rs:581

Functionsimple_broadcast_add

src/cpu/f32/ops.rs:442

Functionsimple_causal_softmax

src/gpu/f32/ops.rs:620

Functionsimple_causal_softmax

src/cpu/f32/ops.rs:534

Functionsimple_causal_softmax

src/cpu/f16/ops.rs:463

Functionsimple_matmul

src/gpu/f32/ops.rs:479

Functionsimple_matmul

src/cpu/f32/ops.rs:450

Functionsimple_matmul

src/cpu/f16/ops.rs:381

Functionsimple_matmul_t

src/gpu/f32/ops.rs:525

Functionsimple_matmul_t

src/cpu/f32/ops.rs:490

Functionsimple_matmul_t

src/cpu/f16/ops.rs:419

Functionsimple_mul

src/gpu/f32/ops.rs:595

Functionsimple_normalize

src/gpu/f32/ops.rs:678

Functionsimple_normalize

src/cpu/f32/ops.rs:590

Functionsimple_normalize

src/cpu/f16/ops.rs:519

Functionsimple_select

src/gpu/f32/ops.rs:665

Functionsimple_select

src/cpu/f32/ops.rs:578

Functionsimple_select

src/cpu/f16/ops.rs:507

Functionsimple_softmax

src/gpu/f32/ops.rs:608

Functionsimple_softmax

src/cpu/f32/ops.rs:523

Functionsimple_softmax

src/cpu/f16/ops.rs:452

Functionsimplify

(x: &mut Tensor)

src/gpu/f32/traits.rs:91

(x: &mut Tensor)

src/cpu/f32/traits.rs:90

Functionspecial_argmax

Argmax of the last dimension of tensor `x `.

src/cpu/f32/ops.rs:367

Functionspecial_argmax

Argmax of the last dimension of tensor `x `.

src/cpu/f16/ops.rs:304

Functionsuggest_setvars_cmd

(x: &mut Tensor)

src/gpu/f32/traits.rs:84

(x: &mut Tensor)

src/cpu/f32/traits.rs:83

Functiontest_cuda_split_heads

src/nn/models/bert.rs:696

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