deepmd.tf.op
This module will house cust Tf OPs after CMake installation.
Submodules
deepmd.tf.op._add_flt_nvnmd_grad
deepmd.tf.op._copy_flt_nvnmd_grad
deepmd.tf.op._dotmul_flt_nvnmd_grad
deepmd.tf.op._flt_nvnmd_grad
deepmd.tf.op._gelu
deepmd.tf.op._map_flt_nvnmd_grad
deepmd.tf.op._matmul_fitnet_nvnmd_grad
deepmd.tf.op._matmul_flt2fix_nvnmd
deepmd.tf.op._matmul_flt_nvnmd_grad
deepmd.tf.op._mul_flt_nvnmd_grad
deepmd.tf.op._prod_force_grad
deepmd.tf.op._prod_force_se_a_grad
deepmd.tf.op._prod_force_se_a_mask_grad
deepmd.tf.op._prod_force_se_r_grad
deepmd.tf.op._prod_virial_grad
deepmd.tf.op._prod_virial_se_a_grad
deepmd.tf.op._prod_virial_se_r_grad
deepmd.tf.op._quantize_nvnmd_grad
deepmd.tf.op._soft_min_force_grad
deepmd.tf.op._soft_min_virial_grad
deepmd.tf.op._tabulate_grad
deepmd.tf.op._tanh4_flt_nvnmd_grad
Package Contents
Functions
Import all custom TF ops that are present in this submodule. |