Expand description
Elementwise kernel signatures
Constantsยง
- ADD_
KERNEL - Add kernel:
c[i] = a[i] + b[i] - GELU_
KERNEL - GELU kernel
- MUL_
KERNEL - Multiply kernel:
c[i] = a[i] * b[i] - RELU_
KERNEL - ReLU kernel:
y[i] = max(0, x[i]) - SIGMOID_
KERNEL - Sigmoid kernel