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
Docs
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