Gradients for tabulate.Module Contents
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)
| |
- deepmd.tf.op._tabulate_grad._tabulate_fusion_se_a_grad_cc(op, dy)[source]
- deepmd.tf.op._tabulate_grad._tabulate_fusion_se_a_grad_grad_cc(op, dy, dy_)[source]
- deepmd.tf.op._tabulate_grad._tabulate_fusion_se_atten_grad_cc(op, dy)[source]
- deepmd.tf.op._tabulate_grad._tabulate_fusion_se_atten_grad_grad_cc(op, dy, dy_, dy_dtwo)[source]
- deepmd.tf.op._tabulate_grad._tabulate_fusion_se_t_grad_cc(op, dy)[source]
- deepmd.tf.op._tabulate_grad._tabulate_fusion_se_t_grad_grad_cc(op, dy, dy_)[source]
- deepmd.tf.op._tabulate_grad._tabulate_fusion_se_r_grad_cc(op, dy)[source]
- deepmd.tf.op._tabulate_grad._tabulate_fusion_se_r_grad_grad_cc(op, dy)[source]