Handling functionals and elementwise ops

Hi,

I was trying to use BERT uncased pretrained model and I noted that functionals and elementwise ops are used instead of class implementation. I was wondering if there is a version of same model available with these rewritten using equivalent class based implementation (Example ops : Add, Mul etc).

Thanks,
gg