This document is relevant for: Trn1, Trn2, Trn3
nki.language.gelu_apprx_sigmoid_dx#
- nki.language.gelu_apprx_sigmoid_dx(x, dtype=None)[source]#
Derivative of sigmoid-approximated GELU, element-wise.
This document is relevant for: Trn1, Trn2, Trn3