Python test_ms_maximum 예제들

예제 #1

def maximum(poly_sch, fuzz_shape=None):
    test_ms_maximum((32, 1024, 1024), (32, 1024, 1024),
                    'float32',
                    poly_sch=poly_sch)
    test_ms_maximum((32, 1024, 1024), (1, 1024, 1024),
                    'float16',
                    poly_sch=poly_sch)
    test_ms_maximum((32, 32, 32, 256), (32, 32, 1, 256),
                    'float16',
                    poly_sch=poly_sch)

예제 #2

def test_maximum():
    test_ms_maximum((32, 1024, 1024), (32, 1024, 1024),
                    'float32',
                    poly_sch=True)
    test_ms_maximum((32, 1024, 1024), (1, 1024, 1024),
                    'float16',
                    poly_sch=True)
    test_ms_maximum((32, 32, 32, 256), (32, 32, 1, 256),
                    'float16',
                    poly_sch=True)
    return True