예제 #1
0
def fused_bn_follow_relu(poly_sch, fuzz_shape=None):
    test_fused_bn_follow_relu((256, 7, 7, 2048),
                              layout='NHWC',
                              poly_sch=poly_sch)
예제 #2
0
def test_ms_fused_bn_follow_relu():
    test_fused_bn_follow_relu((256, 7, 7, 2048), layout='NHWC', poly_sch=True)
    return True