Python test_expand_dims Examples

Example #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)

Example #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