deepmd.tf.op._gelu

Contents

deepmd.tf.op._gelu#

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

Attributes#

Functions#

`_gelu_custom_cc`(→ deepmd.tf.env.tf.Tensor)
`_gelu_grad_custom_cc`(→ list)

Module Contents#

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

deepmd.tf.op._gelu._gelu_custom_cc(op: deepmd.tf.env.tf.Operation, dy: deepmd.tf.env.tf.Tensor) → deepmd.tf.env.tf.Tensor[source]#

deepmd.tf.op._gelu._gelu_grad_custom_cc(op: deepmd.tf.env.tf.Operation, dy: deepmd.tf.env.tf.Tensor) → list[source]#