in include/inplace_abn.h [99:101]
static INLINE_HOST_DEVICE void forward(scalar_t& x, float activation_param) { x = (x >= 0) ? x : static_cast<scalar_t>(x * activation_param); }