Skip to main content

Module elementwise_kernel

Module elementwise_kernel 

Source

Structs§

ElementwiseKernel
elementwise_kernel Kernel
ElementwiseKernelInfo

Functions§

expand
One function of each sample (op, fixed at compile time): 0 |x|, 1 x², 2 √max(x, 0), 3 log₁₀ max(|x|, floor).
launch
Launch the kernel elementwise_kernel() on the given runtime