Source code for deepmd.tf.op._flt_nvnmd_grad

#!/usr/bin/env python3

# SPDX-License-Identifier: LGPL-3.0-or-later
from tensorflow.python.framework import (
    ops,
)

from deepmd.tf.env import (
    op_module,
)


@ops.RegisterGradient("FltNvnmd")


[docs]
def _FltNvnmdGrad(op, grad):
    dx = op_module.flt_nvnmd(grad)
    return [dx]