Python test_fused_bn_update Examples

Example #1

def fused_bn_update(poly_sch, fuzz_shape=None):
    test_fused_bn_update((2048, ), poly_sch=poly_sch)

Example #2

def test_ms_fused_bn_update():
    test_fused_bn_update((2048, ), poly_sch=True)
    return True