deepmd.dpmodel.common

Contents

deepmd.dpmodel.common#

Attributes#

`GLOBAL_NP_FLOAT_PRECISION`
`PRECISION_DICT`
`RESERVED_PRECISON_DICT`
`DEFAULT_PRECISION`

Classes#

The unit operation of a native model.

Module Contents#

deepmd.dpmodel.common.GLOBAL_NP_FLOAT_PRECISION[source]#

deepmd.dpmodel.common.PRECISION_DICT[source]#

deepmd.dpmodel.common.RESERVED_PRECISON_DICT[source]#

deepmd.dpmodel.common.DEFAULT_PRECISION = 'float64'[source]#

class deepmd.dpmodel.common.NativeOP[source]#

Bases: abc.ABC

The unit operation of a native model.

abstract call(*args, **kwargs)[source]#: Forward pass in NumPy implementation.

__call__(*args, **kwargs)[source]#: Forward pass in NumPy implementation.