deepmd.tf.op._tabulate_grad

Gradients for tabulate.

Functions

_tabulate_fusion_se_a_grad_cc(op, dy)
_tabulate_fusion_se_a_grad_grad_cc(op, dy, dy_)
_tabulate_fusion_se_atten_grad_cc(op, dy)
_tabulate_fusion_se_atten_grad_grad_cc(op, dy, dy_, ...)
_tabulate_fusion_se_t_grad_cc(op, dy)
_tabulate_fusion_se_t_grad_grad_cc(op, dy, dy_)
_tabulate_fusion_se_r_grad_cc(op, dy)
_tabulate_fusion_se_r_grad_grad_cc(op, dy)

Module Contents

deepmd.tf.op._tabulate_grad._tabulate_fusion_se_a_grad_cc(op, dy)

deepmd.tf.op._tabulate_grad._tabulate_fusion_se_a_grad_grad_cc(op, dy, dy_)

deepmd.tf.op._tabulate_grad._tabulate_fusion_se_atten_grad_cc(op, dy)

deepmd.tf.op._tabulate_grad._tabulate_fusion_se_atten_grad_grad_cc(op, dy, dy_, dy_dtwo)

deepmd.tf.op._tabulate_grad._tabulate_fusion_se_t_grad_cc(op, dy)

deepmd.tf.op._tabulate_grad._tabulate_fusion_se_t_grad_grad_cc(op, dy, dy_)

deepmd.tf.op._tabulate_grad._tabulate_fusion_se_r_grad_cc(op, dy)

deepmd.tf.op._tabulate_grad._tabulate_fusion_se_r_grad_grad_cc(op, dy)