Functions — github.com/TuckerBMorgan/can-t

CodeHub MCPcopy Create free account

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

Functions825 in github.com/TuckerBMorgan/can-t

⨍ Functions 825◇ Types & classes 48

Functioncat_forward_appends_along_axis

src/central/cat_op.rs:133

Functionchunk_backward_injects_gradients_in_correct_slice

src/central/chunk_op.rs:115

Functionchunk_forward_splits_evenly

src/central/chunk_op.rs:99

Functionclamp_backward_masks_outside_bounds

src/central/clamp_op.rs:71

Functionclamp_forward_handles_fractional_bounds

src/central/clamp_op.rs:58

Functionclip_gradients

(max_norm: f32)

src/central/mod.rs:104

(&self, other: &Self)

src/central/topk_op.rs:32

Functioncomplicated_broadcast_test

src/central/shape.rs:615

Functioncomplicated_matmul_test_shape

src/central/shape.rs:625

Functioncompute_cos_sin_returns_outer_product_shape

src/nn/rotary_embedding.rs:130

Methodcompute_strides_row_major

src/central/shape.rs:498

Functioncos_backward_matches_negative_sine

src/central/cos_op.rs:56

Functioncos_backward_scales_with_incoming_gradient

src/central/cos_op.rs:85

Functioncos_forward_matches_reference_values

src/central/cos_op.rs:71

Methodcreate_tensor_data_and_shape_and_operation

Utility function for create a tensor with data, shape and operation # Arugments 'shape' - the shape of the tensor 'data' - the data

src/central/tensor.rs:283

Functiondecode_gpt2_tokens

src/nn/gpt2.rs:178

src/optimizers/adamw.rs:19

src/optimizers/adam.rs:19

src/central/debugging.rs:14

Functiondiagonal_backward

src/central/diagonal_op.rs:287

Functiondiagonal_backward_batched

src/central/diagonal_op.rs:308

Functiondiagonal_forward_2d_offset

src/central/diagonal_op.rs:237

Functiondiagonal_forward_3d

src/central/diagonal_op.rs:262

Functiondiagonal_forward_batched_axes

src/central/diagonal_op.rs:276

Functiondiagonal_forward_zero_offset

src/central/diagonal_op.rs:251

Functiondiagonal_invalid_offset_panics

src/central/diagonal_op.rs:332

Methoddim_list_to_bitset

(summed_dimensions: &Vec<usize>)

src/central/einsum.rs:5

Functiondimension_length_zero_returns_empty

src/central/topk_op.rs:259

(self, rhs: Self)

src/central/mul_op.rs:47

Functiondiv_f32_test

src/central/mul_op.rs:231

Functiondo_not_add_tensors_of_different_size_but_same_length

src/central/add_op.rs:188

Functiondo_not_mul_tensors_of_different_size_but_same_length

src/central/mul_op.rs:194

Adapated from https://github.com/pytorch/pytorch/blob/0e083942ccd98c5a2d9386f0b2c4bbc901be40ef/aten/src/ATen/native/Linear.cpp einsum function

src/central/einsum.rs:160

Allocates a new tensor with provided shape, where each element is element # Arugments 'shape' - The shape of the allocated tensor 'element' - The valu

src/central/tensor.rs:229

Functionelement_test

src/central/tensor.rs:409

(&self, other: &Self)

src/central/topk_op.rs:16

Functionf32_mul_test

src/central/mul_op.rs:249

(&mut self, mut inputs: Tensor)

src/nn/relu.rs:13

(&mut self, x: Tensor)

src/nn/gelu.rs:13

(&mut self, x: Tensor)

src/nn/swiglu.rs:17

(&mut self, inputs: Tensor)

examples/gpt-oss.rs:87

(modules: Vec<Box<dyn Layer>>)

src/nn/sequential.rs:35

(value: [usize; 1])

src/central/index.rs:15

Methodfrom_gguf_file

( gguf_file: &mut GGUFFile, block_count: i64, config: &GPT2Config, )

src/nn/multihead_attention.rs:51

Methodfrom_gguf_file

Creates a Linear layer from a GGUF file by loading weight and bias tensors # Arguments `weight_tensor_name` - The name of the weight tensor in the GGU

src/nn/linear.rs:37

Methodfrom_gguf_file

(gguf_file: &mut GGUFFile)

src/nn/gpt2.rs:29

Methodfrom_gguf_file

( gguf_file: &mut GGUFFile, block_count: i64, gpt_config: &GPT2Config, )

src/nn/gpt_block.rs:54

Methodfrom_gguf_file

( gguf_file: &mut GGUFFile, weight_name: String, bias_name: String, )

src/nn/layer_norm.rs:21

Methodfrom_gguf_file

Loads a tensor from a GGUF file # Arugments 'tensor_name' - The name of the tensor in the file 'gguf_file' - the data struct that wraps the file we wi

src/central/tensor.rs:182

Methodfrom_tensor

(mut weights: Tensor)

src/nn/embedding.rs:16

Methodfrom_tensors

(mut weights: Tensor, mut bias: Option<Tensor>)

src/nn/linear.rs:24

Creates a tensor from a vector of data with specified shape # Arguments 'data' - The vector of f32 values to populate the tensor 'shape_dims' - The di

src/central/tensor.rs:258

Functiongather_backward_accumulates_duplicate_indices

src/central/gather_op.rs:392

Functiongather_backward_dimension_two_accumulates_counts

src/central/gather_op.rs:228

Functiongather_backward_leaves_unselected_elements_at_zero

src/central/gather_op.rs:262

Functiongather_backward_one_dimensional_tensor_accumulates_duplicates

src/central/gather_op.rs:280

Functiongather_backward_row_dimension_scatter_adds

src/central/gather_op.rs:373

Functiongather_handles_middle_dimension_on_three_dimensional_input

src/central/gather_op.rs:195

Functiongather_handles_one_dimensional_inputs

src/central/gather_op.rs:179

Functiongather_handles_three_dimensional_inputs

src/central/gather_op.rs:352

Functiongather_panics_when_dimension_out_of_range

src/central/gather_op.rs:299

Functiongather_panics_when_index_rank_differs_from_input

src/central/gather_op.rs:308

Functiongather_preserves_shape_when_index_matches_input_rank

src/central/gather_op.rs:164

Functiongather_respects_column_indices

src/central/gather_op.rs:334

Functiongather_respects_row_indices

src/central/gather_op.rs:316

Functiongather_shuffles_columns_when_dimension_is_one

src/central/gather_op.rs:147

Functiongather_swaps_rows_when_index_targets_dimension_zero

src/central/gather_op.rs:130

Methodget_parameters

src/nn/multihead_attention.rs:209

Methodget_parameters

src/nn/gpt2.rs:157

Methodget_parameters

src/nn/relu.rs:17

Methodget_parameters

src/nn/gelu.rs:26

Methodget_parameters

src/nn/tanh.rs:17

Methodget_parameters

src/nn/swiglu.rs:43

Methodget_parameters

examples/autoencoder.rs:131

Methodget_parameters

examples/gpt-oss.rs:115

Methodgpt2_small

src/nn/gpt2.rs:17

Functionk_equals_dimension_length_returns_all_sorted_per_column

src/central/topk_op.rs:296

Functionk_zero_returns_empty

src/central/topk_op.rs:243

Functionl1_loss_backward_produces_sign_gradients

src/central/l1_loss.rs:93

Functionl1_loss_backward_respects_mean_reduction_factor

src/central/l1_loss.rs:135

Functionl1_loss_backward_zero_difference_leaves_grads_zero

src/central/l1_loss.rs:115

Functionl1_loss_is_zero_when_tensors_equal

src/central/l1_loss.rs:57

Functionl1_loss_matches_manual_mae_for_batched_tensor

src/central/l1_loss.rs:69

Functionl1_loss_matches_manual_mae_for_vector

src/central/l1_loss.rs:32

Functionl1_loss_reduces_across_all_dimensions

src/central/l1_loss.rs:44

Methodlabel_to_subscript

src/central/einsum.rs:166

Functionlinear_index_detects_stride_overflow

src/central/gather_op.rs:122

Functionlinear_index_refuses_overflowing_multiplication

src/central/gather_op.rs:110

Functionlinear_index_rejects_dimension_mismatch

src/central/gather_op.rs:100

Functionlinear_index_rejects_out_of_bounds_points

src/central/gather_op.rs:105

Functionlinear_index_respects_row_major_layout

src/central/gather_op.rs:95

Functionlinear_index_returns_expected_value_for_three_dimensions

src/central/gather_op.rs:117

Functionln_f_hook

debug_gradients.py:34

Methodload_data_basic

src/utils/data_set.rs:33

Functionlog_1d_backward_test

src/central/log.rs:204

Functionlog_2d_backward_test

src/central/log.rs:230

Functionlog_2d_test

src/central/log.rs:69

Functionlog_3d_backward_test

src/central/log.rs:257

Functionlog_3d_test

src/central/log.rs:93

Functionlog_4d_test

src/central/log.rs:114

← previous next →301–400 of 825, ranked by callers