def METHOD_NAME(test_case):
    with test_case.assertRaises(TypeError) as context:
        dy = flow.randn(10, 10, dtype=flow.float64)
        label = flow.randn(10, 10, 5, dtype=flow.float32)
        prob = flow.randn(10, 10, 5, dtype=flow.float64)
        flow._C.softmax_cross_entropy_grad(dy, label, prob)
    test_case.assertTrue(
        "label and prob are expected to have the same dtype, but found"
        in str(context.exception)
    )