@ops.RegisterGradient('Unbatch')
def _UnbatchGrad(op, grad):
    return [gen_batch_ops.unbatch_grad(op.inputs[0], op.inputs[1], grad, op.inputs[2], shared_name='unbatch_gradient_{}'.format(op.name)), None, None]
@ops.RegisterGradient('Unbatch')
def _UnbatchGrad(op, grad):
    return [gen_batch_ops.unbatch_grad(op.inputs[0], op.inputs[1], grad, op.inputs[2], shared_name='unbatch_gradient_{}'.format(op.name)), None, None]