Python test_ms_sub 예제들

예제 #1

def sub(poly_sch, fuzz_shape=None):
    test_ms_sub((32, 1024, 1024), (32, 1024, 1024),
                'float32',
                poly_sch=poly_sch)
    test_ms_sub((32, 1024, 1024), (32, 1024, 1024),
                'float16',
                poly_sch=poly_sch)
    test_ms_sub((32, 1024, 1024), (1, 1024, 1024),
                'float32',
                poly_sch=poly_sch)
    test_ms_sub((4, 4, 4), (1, 4, 4), 'float32', poly_sch=poly_sch)

예제 #2

def test_sub():
    test_ms_sub((32, 1024, 1024), (32, 1024, 1024), 'float32', poly_sch=True)
    test_ms_sub((32, 1024, 1024), (32, 1024, 1024), 'float16', poly_sch=True)
    test_ms_sub((32, 1024, 1024), (1, 1024, 1024), 'float32', poly_sch=True)
    test_ms_sub((4, 4, 4), (1, 4, 4), 'float32', poly_sch=True)
    return True