c94e6ba4b3b3cc117d898cb22883fb2a33d5c9a7,tests/python/relay/test_dataflow_pattern.py,,test_partition_dominator,#,705

Before Change

    i = relay.Var("input")
    w = relay.Var("weight")
    f = relay.Function([i, w], generate_diamond(i, w))
    assert tvm.ir.structural_equal(partitioned, f(inp*inp, weight*weight))

def test_quadruple_partition_dominator():
    // Pattern

After Change

    
    i = relay.Var("input")
    w = relay.Var("weight")
    f = relay.Function([i, w], generate_diamond(i, w)).with_attr("PartitionedFromPattern","nn.conv2d_nn.relu_nn.relu_nn.leaky_relu_add_")
    assert tvm.ir.structural_equal(partitioned, f(inp*inp, weight*weight))

def test_quadruple_partition_dominator():
    // Pattern

Instances