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

Attributes

NOT_LOADABLE
PACKAGE_BASE
log

Functions

import_ops()

Import all custom TF ops that are present in this submodule.

Package Contents

deepmd.tf.op.NOT_LOADABLE = ('__init__.py',)

deepmd.tf.op.PACKAGE_BASE = 'deepmd.tf.op'

deepmd.tf.op.log

deepmd.tf.op.import_ops()

Import all custom TF ops that are present in this submodule.

Notes

Initialy this subdir is unpopulated. CMake will install all the op module python files and shared libs.