The mynet from huangshiqing

mynet's People

Contributors

Stargazers

Watchers

mynet's Issues

关于tool/tvm/relay_reverse.py中的line的分类

你好，你开发的这个工具很棒，非常感谢~

有一点想确认一下，在tool/tvm/relay_reverse.py的这个对relay.txt每一行分类的代码中，type=2的情况应该是当前行不包含(的情况才对，因此是不是应该修改成# 2: "%7 = %3.0" ~~or "%99 = (%97, %98);"~~才对呢？看上去%99 = (%97, %98);"像是应该被放在type=1的case。

        for line in f:
            # net_inputs = list()
            # line = "  %114 = subtract(1f /* ty=float32 */, %90) /* ty=Tensor[(1, 12, 50), float32] */;"
            # step 0. get type
            # -1: unkonw
            # 0: "def @main(%INPUT__0..."
            # 1: "%0 = (%INPUT__0, %INPUT__1)"
            # 2: "%7 = %3.0" or "%99 = (%97, %98);"
            # 3: "%9 = nn.scale("
            # 4: "nn.fully_connected(%144..."
            # 5: "}"
            type = -1
            if "def" in line:
                type = 0
            elif "}" in line:
                type = 5
            else:
                index = line.find('(')
                if index == -1:
                    type = 2
                else:
                    index2 = line.find('=')
                    if index-index2 == 2:
                        type = 1
                    else:
                        if line[2] != '%':
                            type = 4
                        else:
                            type = 3

Recommend Projects