Source code for onnx_diagnostic.reference.ops.op_gather_grad

import numpy as np
from onnx.reference.op_run import OpRun
from onnx.reference.ops.op_scatternd import _scatter_nd_impl




class GatherGrad(OpRun):
    op_domain = "com.microsoft"

    def _run(self, shape, indices, updates, reduction=None):
        data = np.zeros(shape, dtype=updates.dtype)
        y = _scatter_nd_impl(data, indices, updates, reduction=reduction)
        return (y,)