Expand description
Hodu CPU Kernels
High-performance CPU kernels for tensor operations with support for exotic floating-point formats and comprehensive integer types.
Modules§
- abs
- add
- add_
scalar - all
- any
- argmax
- argmin
- concat
- conv1d
- conv1d_
grad_ weight - conv2d
- conv2d_
grad_ weight - conv3d
- conv3d_
grad_ weight - conv_
transpose1d - conv_
transpose1d_ grad_ weight - conv_
transpose2d - conv_
transpose2d_ grad_ weight - conv_
transpose3d - conv_
transpose3d_ grad_ weight - cos
- div
- div_
scalar - dot
- eq
- eq_
scalar - exp
- exp2
- exp10
- gather
- ge
- ge_
scalar - gelu
- gt
- gt_
scalar - index_
put - index_
select - le
- le_
scalar - ln
- log2
- log10
- logical_
and - logical_
not - logical_
or - logical_
xor - lt
- lt_
scalar - macros
- matmul
- max
- maximum
- maximum_
scalar - mean
- min
- minimum
- minimum_
scalar - mish
- mul
- mul_
scalar - ne
- ne_
scalar - neg
- norm
- ops_
binary - Binary operations for tensor element-wise computations
- ops_
concat_ split - Concatenation and split operations for tensors
- ops_
conv - Convolution operations for neural network layers
- ops_
indexing - Indexing operations for tensor element selection and manipulation
- ops_
matrix - Matrix operations for linear algebra computations
- ops_
reduce - Reduction operations for aggregating tensor values along dimensions
- ops_
unary - Unary tensor operations
- ops_
windowing - Windowing operations for tensors
- pow
- pow_
scalar - prod
- recip
- reduce_
window_ max - reduce_
window_ mean - reduce_
window_ min - reduce_
window_ sum - relu
- scatter
- scatter_
add - scatter_
max - scatter_
min - sigmoid
- sign
- silu
- sin
- softplus
- split
- sqrt
- square
- std
- sub
- sub_
scalar - sum
- tan
- tanh
- var
Structs§
Enums§
- CpuKernel
Error - Error type for CPU kernel operations
Functions§
- call_
ops_ binary - Execute a binary operation on two tensors
- call_
ops_ concat - Execute a concatenation operation on multiple tensors
- call_
ops_ conv - Execute a convolution operation
- call_
ops_ conv_ grad_ weight - Execute a convolution gradient weight operation
- call_
ops_ dot - Execute a 2D matrix multiplication operation
- call_
ops_ gather - Execute a gather operation
- call_
ops_ index_ put - Execute an index_put operation
- call_
ops_ index_ select - Execute an index_select operation
- call_
ops_ matmul - Execute a batched matrix multiplication operation with broadcasting
- call_
ops_ reduce - Execute a reduction operation
- call_
ops_ reduce_ window - Call windowing reduction operation by kernel name
- call_
ops_ scatter - Execute a scatter operation (scatter, scatter_add, scatter_max, scatter_min)
- call_
ops_ split - Execute a split operation on a tensor
- call_
ops_ unary - Call unary operation by kernel name
- call_
ops_ unary_ scalar - Call unary operation with scalar by kernel name
Type Aliases§
- Result
- Result type for CPU kernel operations