composable_kernel
composable_kernel copied to clipboard
Implement activation other then RELU in Conv + bias + Activation
In CK for (conv + bias + activation) we currently have activation as RELU.
It would be great to have other activations. We can start with the ones that take less performance hit on CK.
eg: TANH, SOFTRELU