nll_loss_op_kernel.cu
3.65 KB
-
Add HardSigmoid && HardSwish && Swish Operator · 9ca4b60fSummary: This commit adds the hardsigmoid, hardswish and swish op with specialized kernel, there are widely used in MobileNetV3 and EfficientNet.
Ting PAN committed