Esempi in Python per test_ms_reshape

Esempio n. 1

def reshape(poly_sch, fuzz_shape=None):
    test_ms_reshape("float32", (64, 128, 1024), (8192, 1024),
                    poly_sch=poly_sch)
    test_ms_reshape("float16", (64, 128, 1024), (8192, 1024),
                    poly_sch=poly_sch)

Esempio n. 2

def test_reshape():
    test_ms_reshape("float32", (64, 128, 1024), (8192, 1024), poly_sch=True)
    test_ms_reshape("float16", (64, 128, 1024), (8192, 1024), poly_sch=True)
    return True