deepmd.tf.op._gelu

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

Module Contents

Functions

_gelu_custom_cc(op, dy)

_gelu_grad_custom_cc(op, dy)

Attributes

gelu

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]