Esempi in Python per test_expand_dims

Esempio n. 1

def expand_dims(poly_sch, fuzz_shape=None):
    test_expand_dims((32, 1024, 1024), 1, 'float32', poly_sch=poly_sch)
    test_expand_dims((32, 1024, 1024), 2, 'float16', poly_sch=poly_sch)

Esempio n. 2

def test_ms_expand_dims():
    test_expand_dims((32, 1024, 1024), 1, 'float32', poly_sch=True)
    test_expand_dims((32, 1024, 1024), 2, 'float16', poly_sch=True)
    return True