deepmd.tf.op._gelu

First-order derivatives and second-order derivatives for gelu function.

Attributes

Functions

`_gelu_custom_cc`(op, dy)
`_gelu_grad_custom_cc`(op, dy)

Module Contents

deepmd.tf.op._gelu.gelu[source]

deepmd.tf.op._gelu._gelu_custom_cc(op, dy)[source]

deepmd.tf.op._gelu._gelu_grad_custom_cc(op, dy)[source]