81ad18edc7360dc7110a479e051ba9161a25f2cf,tests/python/relay/test_pass_merge_composite.py,,make_bn_relu_pattern,#,133

Before Change

         |
       relu
    
    x = relay.var("x")
    gamma = relay.var("gamma")
    beta = relay.var("beta")
    moving_mean = relay.var("moving_mean")
    moving_var = relay.var("moving_var")
    bn_node = relay.nn.batch_norm(x, gamma, beta, moving_mean, moving_var)
    tuple_get_item_node = bn_node[0]
    r = relay.nn.relu(tuple_get_item_node)
    return r


def test_simple_merge():

After Change

         |
       relu
    
    x = wildcard()
    gamma = wildcard()
    beta = wildcard()
    moving_mean = wildcard()
    moving_var = wildcard()
    bn_node = is_op("nn.batch_norm")(x, gamma, beta, moving_mean, moving_var)
    tuple_get_item_node = TupleGetItemPattern(bn_node, 0)
    r = is_op("nn.relu")(tuple_get_item_node)
    return r

def check_result(pattern_table, graph, expected_graph):
    Utility function to check merge composite results.

Instances