Function _ScatterAddGrad

tf_euler/python/euler_ops/mp_ops.py:47–49

(op, grad)

@tf.RegisterGradient('MPScatterAdd')
def _ScatterAddGrad(op, grad):
    indices = op.inputs[1]
    return [gather(grad, indices), None, None]


@tf.RegisterGradient('MPScatterMax')