This document is relevant for: Trn1, Trn2, Trn3

nki.language.gelu_apprx_tanh#

nki.language.gelu_apprx_tanh(x, dtype=None)[source]#

GELU approximation using tanh, element-wise.

This document is relevant for: Trn1, Trn2, Trn3