Module elementwise 

Element-wise operation kernels for GPU

These kernels implement basic element-wise operations that are fundamental to tensor computations: addition, multiplication, division, subtraction.

Structs

ElementwiseAddKernel

Element-wise addition kernel (a + b)

ElementwiseDivKernel

Element-wise division kernel (a / b)

ElementwiseExpKernel

Element-wise exponential kernel (exp(a))

ElementwiseLogKernel

Element-wise logarithm kernel (log(a))

ElementwiseMulKernel

Element-wise multiplication kernel (a * b)

ElementwisePowKernel

Element-wise power kernel (pow(a, b))

ElementwiseSqrtKernel

Element-wise square root kernel (sqrt(a))

ElementwiseSubKernel

Element-wise subtraction kernel (a - b)

ScalarMulKernel

Scalar multiplication kernel (a * scalar)