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

Functiontest_cuda_unsplit_heads

src/nn/models/bert.rs:715

Functiontest_embedding

src/nn/layers/embedding.rs:34

Functiontest_embedding_errors

src/nn/layers/embedding.rs:42

Functiontest_layer_norm

src/nn/layers/layer_norm.rs:38

Functiontest_linear

src/nn/layers/linear.rs:93

Functiontest_split_heads

src/nn/models/bert.rs:674

Functiontest_unsplit_heads

src/nn/models/bert.rs:685

src/nn/layers/embedding.rs:22

(&self, shape: Vec<usize>)

src/gpu/f32/traits.rs:23

(&self, shape: Vec<usize>)

src/cpu/f32/traits.rs:22

Creates a new nulled tensor with given shape ``` use smelte_rs::cpu::f32::Tensor; let tensor = Tensor::zeros(vec![2, 2]); ```

src/cpu/f32/tensor.rs:68

Creates a new nulled tensor with given shape ``` use smelte-rs::cpu::f16::Tensor; let tensor = Tensor::zeros(vec![2, 2]); ```

src/cpu/f16/tensor.rs:53

← previous201–212 of 212, ranked by callers