deepmd.dpmodel.common

Module Contents

Classes

NativeOP

The unit operation of a native model.

Attributes

GLOBAL_NP_FLOAT_PRECISION

PRECISION_DICT

RESERVED_PRECISON_DICT

DEFAULT_PRECISION

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.