You should see both calls, first to selu, and then to elu. Our implementation of selu just calls elu (with a scale parameter, that’s why you don’t see a mul). As long as you only do inference you should be able to register a kernel for selu and intercept that first call. If you want to do Autograd, it gets a bit more complicated.