def METHOD_NAME():
    bb = relax.BlockBuilder()
    g = relax.Var("g", R.Tensor((3, 10, 10)))
    x = relax.Var("x", R.Tensor((3, 5, 10, 10), "float32"))
    y = relax.Var("y", R.Tensor((3, 10, 10), "int64"))
    w = relax.Var("w", R.Tensor((5,), "float32"))
    _check_inference(bb, relax.op.grad.nll_loss_backward(g, x, y), x.struct_info)
    _check_inference(bb, relax.op.grad.nll_loss_backward(g, x, y, w), x.struct_info)