Function _GatherGrad

tf_euler/python/euler_ops/mp_ops.py:40–43

(op, grad)

@tf.RegisterGradient('MPGather')
def _GatherGrad(op, grad):
    params = op.inputs[0]
    indices = op.inputs[1]
    return [scatter_add(grad, indices, tf.shape(params)[0]), None]


@tf.RegisterGradient('MPScatterAdd')