This document is relevant for: Trn1, Trn2, Trn3

nki.language.gelu_apprx_sigmoid#

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

GELU approximation using sigmoid, element-wise.

This document is relevant for: Trn1, Trn2, Trn3