MCPcopy Create free account

hub / github.com/DragonflyRobotics/Neuroxide / functions

Functions180 in github.com/DragonflyRobotics/Neuroxide

⨍ Functions 180◇ Types & classes 39
↓ 299 callersFunctionnew
(b: &mut test::Bencher)
bench/bench_tensor.rs:10
↓ 138 callersMethodclone
(&self)
crates/mempool/src/pointers.rs:15
↓ 72 callersMethodget_shape
(&self)
crates/types/src/tensor.rs:86
↓ 17 callersMethodget_branches
(&self)
crates/types/src/cat.rs:106
↓ 17 callersMethodlock_ref
(&self)
crates/types/src/tensor_element.rs:29
↓ 14 callersMethodbackward
(&mut self)
crates/types/src/tensor.rs:98
↓ 13 callersMethodset_gradient
(&mut self, grad: SharedTensor<T>)
crates/types/src/tensor.rs:106
↓ 12 callersMethodget_stride
(&self)
crates/types/src/tensor.rs:114
↓ 12 callersMethodvalues_clone
(&self)
crates/types/src/tensor_element.rs:33
↓ 11 callersMethodget_values_slice
(&self)
crates/types/src/tensor.rs:82
↓ 11 callersMethodinto_inputs
(self)
crates/types/src/input.rs:8
↓ 10 callersFunctionforward
(b: &mut test::Bencher)
bench/ops/bench_ln.rs:8
↓ 7 callersMethodget_values
(&self)
crates/types/src/tensor.rs:79
↓ 6 callersFunctionforward
(b: &mut test::Bencher)
bench/ops/bench_mul.rs:8
↓ 6 callersFunctionforward
(b: &mut test::Bencher)
bench/ops/bench_add.rs:8
↓ 6 callersFunctionforward
(b: &mut test::Bencher)
bench/ops/bench_sub.rs:8
↓ 6 callersFunctionforward
(b: &mut test::Bencher)
bench/ops/bench_matmul.rs:9
↓ 6 callersFunctionnew
(b: &mut test::Bencher)
bench/bench_tensordb.rs:8
↓ 5 callersMethodas_ptr
(&self)
crates/mempool/src/dummyptr.rs:22
↓ 5 callersMethodfree
(&mut self, _ptr: *mut u8)
crates/mempool/src/dummydevice.rs:16
↓ 4 callersMethodforward
(inputs: I)
crates/ops/src/op.rs:23
↓ 3 callersMethodcmp
(&self, other: &Self)
crates/mempool/src/block.rs:47
↓ 3 callersFunctionforward
(b: &mut test::Bencher)
bench/ops/bench_pow.rs:8
↓ 3 callersFunctionforward
(b: &mut test::Bencher)
bench/ops/bench_div.rs:8
↓ 3 callersMethodget_prev
(&self)
crates/mempool/src/block.rs:136
↓ 3 callersMethodmalloc
(&mut self, size: usize)
crates/mempool/src/dummydevice.rs:10
↓ 3 callersMethodoffset
(&self, offset: usize)
crates/mempool/src/dummyptr.rs:34
↓ 3 callersMethodprint
(&self)
crates/mempool/src/pool.rs:98
↓ 3 callersMethodset_op
(&mut self, op: Arc<Mutex<dyn OperationStub<T>>>)
crates/types/src/tensor.rs:75
↓ 3 callersMethodsum
(one: &SharedTensor<T>, axis: usize)
crates/types/src/tensor.rs:267
↓ 2 callersMethodcat
(one: &SharedTensor<T>, other: &SharedTensor<T>, axis: usize)
crates/types/src/tensor.rs:126
↓ 2 callersFunctionforward
(b: &mut test::Bencher)
bench/ops/bench_sin.rs:8
↓ 2 callersFunctionforward
(b: &mut test::Bencher)
bench/ops/bench_cos.rs:8
↓ 2 callersMethodfree
(&mut self, block: &BlockRef)
crates/mempool/src/pool.rs:72
↓ 2 callersMethodget_gradient
(&self)
crates/types/src/tensor.rs:110
↓ 2 callersMethodget_next
(&self)
crates/mempool/src/block.rs:133
↓ 2 callersMethodinto_shape
(self)
crates/types/src/input.rs:51
↓ 2 callersMethodmalloc
(&mut self, size: usize)
crates/mempool/src/pool.rs:46
↓ 2 callersMethodpermute
(one: &SharedTensor<T>, reordered_axis: Box<[usize]>)
crates/types/src/tensor.rs:472
↓ 1 callersMethodallocate
(&mut self, size: usize)
crates/mempool/src/dummydevice.rs:25
↓ 1 callersMethodas_ptr
(&self)
crates/cuda/src/memory/cudaptr.rs:22
↓ 1 callersFunctionbatched_sgemm
Batched GEMM for tensors with arbitrary batch dims A shape: [..., M, K] B shape: [..., K, N] Returns flattened output: [..., M, N]
crates/ops/src/matmul.rs:64
↓ 1 callersFunctionbench_pool_alloc_free_stress
()
bin/src/main.rs:10
↓ 1 callersFunctionbench_torch_alloc_free_stress
( iters=50_000, min_bytes=1 * 1024, max_bytes=128 * 1024 * 1024, keep_prob=0.1, )
stress.py:7
↓ 1 callersMethoddeallocate
(&mut self, ptr: Box<dyn DevicePtr>)
crates/mempool/src/dummydevice.rs:30
↓ 1 callersFunctionforward_cuda
(b: &mut test::Bencher)
bench/ops/bench_ln.rs:19
↓ 1 callersMethodget_device
(&self)
crates/types/src/tensor.rs:94
↓ 1 callersMethodget_op
(&self)
crates/types/src/tensor.rs:90
↓ 1 callersMethodinto_values
(self)
crates/types/src/input.rs:79
↓ 1 callersMethodjoin_next
(&self, heap: &mut BTreeSet<BlockRef>)
crates/mempool/src/block.rs:100
↓ 1 callersMethodjoin_prev
(&self, heap: &mut BTreeSet<BlockRef>)
crates/mempool/src/block.rs:115
↓ 1 callersMethodoffset
(&self, offset: usize)
crates/cuda/src/memory/cudaptr.rs:34
↓ 1 callersFunctionsgemm_helper
Single GEMM call (row-major, no transpose)
crates/ops/src/matmul.rs:30
↓ 1 callersMethodshift
(&self, offset: isize)
crates/mempool/src/dummyptr.rs:12
↓ 1 callersMethodshift
(&self, offset: isize)
crates/cuda/src/memory/cudaptr.rs:12
↓ 1 callersMethodsplit
(self: &BlockRef, split_size: usize)
crates/mempool/src/block.rs:76
↓ 1 callersMethodsqueeze
(one: &SharedTensor<T>, axis: usize)
crates/types/src/tensor.rs:419
↓ 1 callersMethodunsqueeze
(one: &SharedTensor<T>, axis: usize)
crates/types/src/tensor.rs:397
Methodadd
(self, offset: usize)
crates/mempool/src/pointers.rs:23
Methodallocate
(&mut self, size: usize)
pool.rs:24
Methodallocate
(&mut self, size: usize)
crates/cuda/src/memory/cudadevice.rs:14
Methodapply_grad
(tensor: SharedTensor<T>, grad: SharedTensor<T>)
crates/ops/src/op.rs:11
Functionbackward
(b: &mut test::Bencher)
bench/ops/bench_mul.rs:42
Functionbackward
(b: &mut test::Bencher)
bench/ops/bench_add.rs:42
Functionbackward
(b: &mut test::Bencher)
bench/ops/bench_pow.rs:42
Functionbackward
(b: &mut test::Bencher)
bench/ops/bench_sub.rs:42
Functionbackward
(b: &mut test::Bencher)
bench/ops/bench_sin.rs:30
Functionbackward
(b: &mut test::Bencher)
bench/ops/bench_ln.rs:30
Functionbackward
(b: &mut test::Bencher)
bench/ops/bench_div.rs:42
Functionbackward
(b: &mut test::Bencher)
bench/ops/bench_matmul.rs:62
Functionbackward
(b: &mut test::Bencher)
bench/ops/bench_cos.rs:30
Methodbackward
(&mut self, upstream: SharedTensor<T>)
crates/ops/src/add.rs:103
Methodbackward
(&mut self, upstream: SharedTensor<T>)
crates/ops/src/matmul.rs:148
Methodbackward
(&mut self, upstream: SharedTensor<T>)
crates/ops/src/mul.rs:105
Methodbackward
(&mut self, upstream: SharedTensor<T>)
crates/types/src/cat.rs:30
Methodbackward
(&mut self, upstream: SharedTensor<T>)
crates/types/src/broadcast.rs:36
Methodbackward
(&mut self, upstream: SharedTensor<T>)
crates/types/src/view.rs:30
Methodbackward
(&mut self, upstream: SharedTensor<T>)
crates/types/src/slice.rs:30
Methodbackward
(&mut self, upstream: SharedTensor<T>)
crates/types/src/unsqueeze.rs:30
Methodbackward
(&mut self, upstream: SharedTensor<T>)
crates/types/src/axis_sum.rs:30
Methodbackward
(&mut self, upstream: SharedTensor<T>)
crates/types/src/permute.rs:31
Methodbackward
(&mut self, upstream: SharedTensor<T>)
crates/types/src/squeeze.rs:30
Methodbackward
(&self)
crates/types/src/tensor_element.rs:40
Methodbroadcast_linear
( one: &SharedTensor<T>, other: &SharedTensor<T>, )
crates/types/src/tensor.rs:524
Methodbroadcast_matmul
( one: &SharedTensor<T>, other: &SharedTensor<T>, )
crates/types/src/tensor.rs:732
Methodbroadcast_shapes_linear
( shape1: &mut Vec<usize>, shape2: &mut Vec<usize>, stride1: &mut Vec<usize>,
crates/types/src/tensor.rs:216
Methodcheck_forward_inputs
( inputs: I, )
crates/ops/src/add.rs:18
Methodcheck_forward_inputs
( inputs: I, )
crates/ops/src/matmul.rs:102
Methodcheck_forward_inputs
( inputs: I, )
crates/ops/src/mul.rs:20
Methodcheck_forward_inputs
( inputs: I, )
crates/types/src/cat.rs:15
Methodcheck_forward_inputs
( inputs: I, )
crates/types/src/broadcast.rs:21
Methodcheck_forward_inputs
( inputs: I, )
crates/types/src/view.rs:15
Methodcheck_forward_inputs
( inputs: I, )
crates/types/src/slice.rs:15
Methodcheck_forward_inputs
( inputs: I, )
crates/types/src/unsqueeze.rs:15
Methodcheck_forward_inputs
( inputs: I, )
crates/types/src/axis_sum.rs:15
Methodcheck_forward_inputs
( inputs: I, )
crates/types/src/permute.rs:15
Methodcheck_forward_inputs
( inputs: I, )
crates/types/src/squeeze.rs:15
Functionclear_graph
(b: &mut test::Bencher)
bench/bench_tensor.rs:24
Methodclone
(&self)
crates/mempool/src/dummyptr.rs:30
Methodclone
(&self)
crates/cuda/src/memory/cudaptr.rs:30
next →1–100 of 180, ranked by callers