Exemplos de test_ms_round em Python

Exemplo n.º 1

def round(poly_sch, fuzz_shape=None):
    test_ms_round((1024, 1024), "float32", poly_sch=poly_sch)
    test_ms_round((1024, 1024), "float16", poly_sch=poly_sch)
    test_ms_round((1, ), "float32", poly_sch=poly_sch)
    test_ms_round((1, 1), "float32", poly_sch=poly_sch)
    test_ms_round((1, ), "float16", poly_sch=poly_sch)
    test_ms_round((1, 1), "float16", poly_sch=poly_sch)

Exemplo n.º 2

def test_round():
    test_ms_round((1024, 1024), "float32", poly_sch=True)
    test_ms_round((1024, 1024), "float16", poly_sch=True)
    test_ms_round((1, ), "float32", poly_sch=True)
    test_ms_round((1, 1), "float32", poly_sch=True)
    test_ms_round((1, ), "float16", poly_sch=True)
    test_ms_round((1, 1), "float16", poly_sch=True)
    return True