Functions — github.com/TuckerBMorgan/can-t

hub / github.com/TuckerBMorgan/can-t / functions

Functions825 in github.com/TuckerBMorgan/can-t

Takes two tensors as flat buffers, and preforms matrix multiplication upon them It makes an assumption that all of them are 4d. This is to make things

crates/cant_cpu/src/tensor_matmul.rs:13

↓ 1 callersFunctionvalid_shape

(a: [usize; MAX_DIMS], b: [usize; MAX_DIMS])

crates/cant_metal/src/tensor_matmul.rs:17

↓ 1 callersFunctionvalidate_tensor_store

()

src/central/mod.rs:108

↓ 1 callersMethodvalidate_tensor_store

(&self)

src/central/equation.rs:860

Methodadd

(self, rhs: Self)

src/central/add_op.rs:9

Functionadd_remove_test

()

src/central/shape.rs:600

Methodadd_vector

Takes two tensors as flat buffers, adds them together at an elementwise level and returns the result # Arguments 'a' : The first tensor 'b' : the seco

src/central/equation.rs:275

Functionadvance_pow_test

()

src/central/pow_op.rs:68

Functionadvanced_batch_mutiply_test

()

src/central/einsum.rs:528

Methodall_unique

(values: &[usize])

src/central/permute.rs:68

Functionallocate_and_data_test

()

src/central/tensor.rs:391

Functionallocate_test

()

src/central/tensor.rs:386

Functionapply_rotary_embedding_identity_cos_sin_no_change

()

src/nn/rotary_embedding.rs:147

Functionapply_rotary_embedding_preserves_input_shape

()

src/nn/rotary_embedding.rs:138

Functionapply_rotary_embedding_quadrature_rotation

()

src/nn/rotary_embedding.rs:158

Functionapprox_equal

(a: f32, b: f32, epsilon: f32)

src/lib.rs:12

Functionapprox_equal

(a: f32, b: f32, epsilon: f32)

src/nn/linear.rs:84

Functionapprox_equal

(a: f32, b: f32, epsilon: f32)

src/nn/embedding.rs:37

Functionapprox_equal

(a: f32, b: f32, epsilon: f32)

src/nn/sequential.rs:46

Functionapprox_equal

(a: f32, b: f32, epsilon: f32)

src/nn/gpt_block.rs:139

Functionapprox_equal

(a: f32, b: f32, epsilon: f32)

src/nn/layer_norm.rs:81

Functionapprox_equal

(a: f32, b: f32, epsilon: f32)

src/nn/tanh.rs:26

Functionapprox_equal

(a: f32, b: f32, epsilon: f32)

src/nn/rmsnorm.rs:51

Functionapprox_equal

(a: f32, b: f32, epsilon: f32)

src/central/transpose_op.rs:102

Functionapprox_equal

(a: f32, b: f32, epsilon: f32)

src/central/sigmoid_op.rs:54

Functionapprox_equal

(a: f32, b: f32, epsilon: f32)

src/central/reshape.rs:39

Functionapprox_equal

(a: f32, b: f32, epsilon: f32)

src/central/relu_op.rs:49

Functionapprox_equal

(a: f32, b: f32, epsilon: f32)

src/central/add_op.rs:113

Functionapprox_equal

(a: f32, b: f32, epsilon: f32)

src/central/select_op.rs:125

Functionapprox_equal

(a: f32, b: f32, epsilon: f32)

src/central/matmul_op.rs:146

Functionapprox_equal

(a: f32, b: f32, epsilon: f32)

src/central/std_op.rs:176

Functionapprox_equal

(a: f32, b: f32, epsilon: f32)

src/central/clamp_op.rs:53

Functionapprox_equal

(a: f32, b: f32)

src/central/max_op.rs:91

Functionapprox_equal

(a: f32, b: f32, epsilon: f32)

src/central/mean_op.rs:103

Functionapprox_equal

(a: f32, b: f32)

src/central/gather_op.rs:90

Functionapprox_equal

(a: f32, b: f32, epsilon: f32)

src/central/sin_op.rs:51

Functionapprox_equal

(a: f32, b: f32, epsilon: f32)

src/central/cos_op.rs:51

Functionapprox_equal

(a: f32, b: f32, epsilon: f32)

src/central/stack_op.rs:41

Functionapprox_equal

(a: f32, b: f32, epsilon: f32)

src/central/l1_loss.rs:27

Functionapprox_equal

(a: f32, b: f32, epsilon: f32)

src/central/diagonal_op.rs:232

Functionapprox_equal

(a: f32, b: f32, epsilon: f32)

src/central/tanh_op.rs:49

Functionapprox_equal

(a: f32, b: f32, epsilon: f32)

src/central/log.rs:42

Functionapprox_equal

(a: f32, b: f32, epsilon: f32)

src/central/cat_op.rs:128

Functionapprox_equal

(a: f32, b: f32, epsilon: f32)

src/central/sum_op.rs:99

Functionapprox_equal

(a: f32, b: f32, epsilon: f32)

src/central/pow_op.rs:45

Functionapprox_equal

(a: f32, b: f32, epsilon: f32)

src/central/masked_fill.rs:59

Functionapprox_equal

(a: f32, b: f32, epsilon: f32)

src/central/chunk_op.rs:94

Functionapprox_equal

(a: f32, b: f32, epsilon: f32)

src/central/softmax_op.rs:111

Methodarange

Creates a tensor of a range of elemenbts, based on the parameters # Arguments 'start' - Where we start the range 'length' - The number of steps we wil

src/central/tensor.rs:268

Functionbackprop_add_test

()

src/central/add_op.rs:197

Functionbackprop_mul_test

()

src/central/mul_op.rs:203

Methodbackward

Preforms the backward pass of a value through the network it only sets the grad, it does not zero it out, or update the values # Arugments 'starting_v

src/central/equation.rs:689

Functionbad_broadacst_shape_test

()

src/central/shape.rs:735

Functionbad_broadacst_test

()

src/central/shape.rs:727

Functionbad_can_matmul

()

src/central/shape.rs:696

Functionbad_shape_test

()

src/central/shape.rs:683

Functionbad_shape_test_remove_at_zero

()

src/central/shape.rs:689

Functionbasic

()

src/nn/bpe_tokenizer.rs:97

Functionbasic_1d_1d_test

()

src/central/einsum.rs:513

Functionbasic_2d_broadcast_test

()

src/central/add_op.rs:171

Functionbasic_2d_broadcast_test

()

src/central/mul_op.rs:177

Functionbasic_2d_test

()

src/central/add_op.rs:127

Functionbasic_2d_test

()

src/central/mul_op.rs:144

Functionbasic_3d_brodcast

()

src/central/add_op.rs:149

Functionbasic_3d_test

()

src/central/add_op.rs:138

Functionbasic_3d_test

()

src/central/mul_op.rs:155

Functionbasic_4d_test

()

src/central/add_op.rs:160

Functionbasic_4d_test

()

src/central/mul_op.rs:166

Functionbasic_add_test

()

src/central/add_op.rs:118

Functionbasic_add_test

()

src/central/shape.rs:575

Functionbasic_allocation_test

()

src/central/shape.rs:522

Functionbasic_allocation_test_2

()

src/central/shape.rs:530

Functionbasic_batch_test

()

src/central/einsum.rs:521

Functionbasic_div_test

()

src/central/mul_op.rs:135

Functionbasic_double_index_sum_test

()

src/central/sum_op.rs:123

Functionbasic_double_index_with_skip_sum_test

()

src/central/sum_op.rs:146

Functionbasic_matmul_shape_tests

()

src/central/shape.rs:646

Functionbasic_matmul_test

()

src/central/matmul_op.rs:160

Functionbasic_mul_test

()

src/central/mul_op.rs:117

Functionbasic_neg_test

()

src/central/mul_op.rs:126

Functionbasic_pow_test

()

src/central/pow_op.rs:59

Functionbasic_remove_test

()

src/central/shape.rs:542

Functionbasic_reshape_backward_test

()

src/central/reshape.rs:73

Functionbasic_reshape_test

()

src/central/reshape.rs:53

Functionbasic_reshape_test_2

()

src/central/reshape.rs:64

Functionbasic_retrain_test

()

src/nn/gpt2.rs:283

Functionbasic_select_test

()

src/central/select_op.rs:130

Functionbasic_set_test

()

src/central/index.rs:47

Functionbasic_single_column_k1

()

src/central/topk_op.rs:267

Functionbasic_sum_test

()

src/central/sum_op.rs:104

src/optimizers/stochastic_gradient_descent.rs:65

Functionbasic_test

()

src/optimizers/adamw.rs:134

Functionbasic_test

()

src/optimizers/adam.rs:125

Functionbasic_test

()

src/central/einsum.rs:506

Functionbatch_norm_simple_test

()

src/lib.rs:156

Methodbatch_with_indices

Returns (values, indices), both sized k * num_cols, laid out column-major: for each "column" c in 0..num_cols, the k results live at offsets c + j * n

src/central/topk_op.rs:42

Functioncan_matmul_test

()

src/central/shape.rs:671

Methodcan_reshape_to

checks if you can reshape one shape into another

src/central/shape.rs:433

Functioncat_backward_splits_gradients

()

src/central/cat_op.rs:144

← previous next →201–300 of 825, ranked by callers