InfiniTensor/test/nnet/log/conv2gemm/Conv2gemm_NCHW_RSFC_11.expr

1574 lines
25 KiB
Plaintext

{
"Version": 1,
"0": {
"type": 4,
"name": "T21",
"shape": [
1,
7,
7,
512
],
"paddings": [
0,
0,
0,
0
],
"source": "1"
},
"1": {
"type": 105,
"inputs": [
"2"
],
"expr": "80",
"outputShape": [
1,
7,
7,
512
]
},
"2": {
"type": 4,
"name": "T5",
"shape": [
49,
4608
],
"paddings": [
0,
0
],
"source": "3"
},
"3": {
"type": 101,
"inputs": [
"4",
"20"
],
"expr": "38",
"args": [
1,
49,
4608,
512,
false,
true
]
},
"4": {
"type": 4,
"name": "T4",
"shape": [
49,
512
],
"paddings": [
0,
0
],
"source": "5"
},
"5": {
"type": 105,
"inputs": [
"6"
],
"expr": "7",
"outputShape": [
49,
512
]
},
"6": {
"type": 4,
"name": "A",
"shape": [
1,
7,
7,
512
],
"paddings": [
0,
1,
1,
0
],
"source": "-1"
},
"7": {
"type": 2,
"paddings": [
0,
0
],
"summand": "8",
"loopVarRanges": {
"i32": [
0,
49
],
"c": [
0,
512
]
}
},
"8": {
"type": 3,
"subExprsNum": 4,
"object": "9",
"indexes": [
"10",
"13",
"16",
"19"
]
},
"9": {
"type": 4,
"name": "A",
"shape": [
1,
7,
7,
512
],
"paddings": [
0,
1,
1,
0
],
"source": "-1"
},
"10": {
"type": 1,
"opType": 3,
"lhs": "11",
"rhs": "12"
},
"11": {
"type": 5,
"name": "i32"
},
"12": {
"type": 0,
"val": 49
},
"13": {
"type": 1,
"opType": 3,
"lhs": "14",
"rhs": "15"
},
"14": {
"type": 5,
"name": "i32"
},
"15": {
"type": 0,
"val": 7
},
"16": {
"type": 1,
"opType": 4,
"lhs": "17",
"rhs": "18"
},
"17": {
"type": 5,
"name": "i32"
},
"18": {
"type": 0,
"val": 7
},
"19": {
"type": 5,
"name": "c"
},
"20": {
"type": 4,
"name": "T3",
"shape": [
4608,
512
],
"paddings": [
0,
0
],
"source": "21"
},
"21": {
"type": 105,
"inputs": [
"22"
],
"expr": "23",
"outputShape": [
4608,
512
]
},
"22": {
"type": 4,
"name": "K",
"shape": [
3,
3,
512,
512
],
"paddings": [
0,
0,
0,
0
],
"source": "-1"
},
"23": {
"type": 2,
"paddings": [
0,
0
],
"summand": "24",
"loopVarRanges": {
"i31": [
0,
4608
],
"c": [
0,
512
]
}
},
"24": {
"type": 3,
"subExprsNum": 4,
"object": "25",
"indexes": [
"26",
"29",
"34",
"37"
]
},
"25": {
"type": 4,
"name": "K",
"shape": [
3,
3,
512,
512
],
"paddings": [
0,
0,
0,
0
],
"source": "-1"
},
"26": {
"type": 1,
"opType": 3,
"lhs": "27",
"rhs": "28"
},
"27": {
"type": 5,
"name": "i31"
},
"28": {
"type": 0,
"val": 1536
},
"29": {
"type": 1,
"opType": 4,
"lhs": "30",
"rhs": "33"
},
"30": {
"type": 1,
"opType": 3,
"lhs": "31",
"rhs": "32"
},
"31": {
"type": 5,
"name": "i31"
},
"32": {
"type": 0,
"val": 512
},
"33": {
"type": 0,
"val": 3
},
"34": {
"type": 1,
"opType": 4,
"lhs": "35",
"rhs": "36"
},
"35": {
"type": 5,
"name": "i31"
},
"36": {
"type": 0,
"val": 512
},
"37": {
"type": 5,
"name": "c"
},
"38": {
"type": 2,
"paddings": [
0,
0
],
"summand": "39",
"loopVarRanges": {
"i32": [
0,
49
],
"i31": [
0,
4608
]
},
"sumVarRanges": {
"c": [
0,
512
]
}
},
"39": {
"type": 1,
"opType": 2,
"lhs": "40",
"rhs": "59"
},
"40": {
"type": 3,
"subExprsNum": 2,
"object": "41",
"indexes": [
"57",
"58"
]
},
"41": {
"type": 4,
"name": "T4",
"shape": [
49,
512
],
"paddings": [
0,
0
],
"source": "42"
},
"42": {
"type": 105,
"inputs": [
"43"
],
"expr": "44",
"outputShape": [
49,
512
]
},
"43": {
"type": 4,
"name": "A",
"shape": [
1,
7,
7,
512
],
"paddings": [
0,
1,
1,
0
],
"source": "-1"
},
"44": {
"type": 2,
"paddings": [
0,
0
],
"summand": "45",
"loopVarRanges": {
"i32": [
0,
49
],
"c": [
0,
512
]
}
},
"45": {
"type": 3,
"subExprsNum": 4,
"object": "46",
"indexes": [
"47",
"50",
"53",
"56"
]
},
"46": {
"type": 4,
"name": "A",
"shape": [
1,
7,
7,
512
],
"paddings": [
0,
1,
1,
0
],
"source": "-1"
},
"47": {
"type": 1,
"opType": 3,
"lhs": "48",
"rhs": "49"
},
"48": {
"type": 5,
"name": "i32"
},
"49": {
"type": 0,
"val": 49
},
"50": {
"type": 1,
"opType": 3,
"lhs": "51",
"rhs": "52"
},
"51": {
"type": 5,
"name": "i32"
},
"52": {
"type": 0,
"val": 7
},
"53": {
"type": 1,
"opType": 4,
"lhs": "54",
"rhs": "55"
},
"54": {
"type": 5,
"name": "i32"
},
"55": {
"type": 0,
"val": 7
},
"56": {
"type": 5,
"name": "c"
},
"57": {
"type": 5,
"name": "i32"
},
"58": {
"type": 5,
"name": "c"
},
"59": {
"type": 3,
"subExprsNum": 2,
"object": "60",
"indexes": [
"78",
"79"
]
},
"60": {
"type": 4,
"name": "T3",
"shape": [
4608,
512
],
"paddings": [
0,
0
],
"source": "61"
},
"61": {
"type": 105,
"inputs": [
"62"
],
"expr": "63",
"outputShape": [
4608,
512
]
},
"62": {
"type": 4,
"name": "K",
"shape": [
3,
3,
512,
512
],
"paddings": [
0,
0,
0,
0
],
"source": "-1"
},
"63": {
"type": 2,
"paddings": [
0,
0
],
"summand": "64",
"loopVarRanges": {
"i31": [
0,
4608
],
"c": [
0,
512
]
}
},
"64": {
"type": 3,
"subExprsNum": 4,
"object": "65",
"indexes": [
"66",
"69",
"74",
"77"
]
},
"65": {
"type": 4,
"name": "K",
"shape": [
3,
3,
512,
512
],
"paddings": [
0,
0,
0,
0
],
"source": "-1"
},
"66": {
"type": 1,
"opType": 3,
"lhs": "67",
"rhs": "68"
},
"67": {
"type": 5,
"name": "i31"
},
"68": {
"type": 0,
"val": 1536
},
"69": {
"type": 1,
"opType": 4,
"lhs": "70",
"rhs": "73"
},
"70": {
"type": 1,
"opType": 3,
"lhs": "71",
"rhs": "72"
},
"71": {
"type": 5,
"name": "i31"
},
"72": {
"type": 0,
"val": 512
},
"73": {
"type": 0,
"val": 3
},
"74": {
"type": 1,
"opType": 4,
"lhs": "75",
"rhs": "76"
},
"75": {
"type": 5,
"name": "i31"
},
"76": {
"type": 0,
"val": 512
},
"77": {
"type": 5,
"name": "c"
},
"78": {
"type": 5,
"name": "i31"
},
"79": {
"type": 5,
"name": "c"
},
"80": {
"type": 2,
"paddings": [
0,
0,
0,
0
],
"summand": "81",
"loopVarRanges": {
"n": [
0,
1
],
"h": [
0,
7
],
"w": [
0,
7
],
"f": [
0,
512
]
},
"sumVarRanges": {
"r": [
0,
3
],
"s": [
0,
3
]
}
},
"81": {
"type": 3,
"subExprsNum": 6,
"object": "82",
"indexes": [
"184",
"187",
"188",
"191",
"192",
"193"
]
},
"82": {
"type": 2,
"paddings": [
1,
0,
1,
0,
0,
0
],
"summand": "83",
"loopVarRanges": {
"i27": [
1,
8
],
"i28": [
0,
3
],
"i17": [
1,
8
],
"i18": [
0,
3
],
"n": [
0,
1
],
"f": [
0,
512
]
}
},
"83": {
"type": 3,
"subExprsNum": 2,
"object": "84",
"indexes": [
"162",
"175"
]
},
"84": {
"type": 4,
"name": "T5",
"shape": [
49,
4608
],
"paddings": [
0,
0
],
"source": "85"
},
"85": {
"type": 101,
"inputs": [
"86",
"102"
],
"expr": "120",
"args": [
1,
49,
4608,
512,
false,
true
]
},
"86": {
"type": 4,
"name": "T4",
"shape": [
49,
512
],
"paddings": [
0,
0
],
"source": "87"
},
"87": {
"type": 105,
"inputs": [
"88"
],
"expr": "89",
"outputShape": [
49,
512
]
},
"88": {
"type": 4,
"name": "A",
"shape": [
1,
7,
7,
512
],
"paddings": [
0,
1,
1,
0
],
"source": "-1"
},
"89": {
"type": 2,
"paddings": [
0,
0
],
"summand": "90",
"loopVarRanges": {
"i32": [
0,
49
],
"c": [
0,
512
]
}
},
"90": {
"type": 3,
"subExprsNum": 4,
"object": "91",
"indexes": [
"92",
"95",
"98",
"101"
]
},
"91": {
"type": 4,
"name": "A",
"shape": [
1,
7,
7,
512
],
"paddings": [
0,
1,
1,
0
],
"source": "-1"
},
"92": {
"type": 1,
"opType": 3,
"lhs": "93",
"rhs": "94"
},
"93": {
"type": 5,
"name": "i32"
},
"94": {
"type": 0,
"val": 49
},
"95": {
"type": 1,
"opType": 3,
"lhs": "96",
"rhs": "97"
},
"96": {
"type": 5,
"name": "i32"
},
"97": {
"type": 0,
"val": 7
},
"98": {
"type": 1,
"opType": 4,
"lhs": "99",
"rhs": "100"
},
"99": {
"type": 5,
"name": "i32"
},
"100": {
"type": 0,
"val": 7
},
"101": {
"type": 5,
"name": "c"
},
"102": {
"type": 4,
"name": "T3",
"shape": [
4608,
512
],
"paddings": [
0,
0
],
"source": "103"
},
"103": {
"type": 105,
"inputs": [
"104"
],
"expr": "105",
"outputShape": [
4608,
512
]
},
"104": {
"type": 4,
"name": "K",
"shape": [
3,
3,
512,
512
],
"paddings": [
0,
0,
0,
0
],
"source": "-1"
},
"105": {
"type": 2,
"paddings": [
0,
0
],
"summand": "106",
"loopVarRanges": {
"i31": [
0,
4608
],
"c": [
0,
512
]
}
},
"106": {
"type": 3,
"subExprsNum": 4,
"object": "107",
"indexes": [
"108",
"111",
"116",
"119"
]
},
"107": {
"type": 4,
"name": "K",
"shape": [
3,
3,
512,
512
],
"paddings": [
0,
0,
0,
0
],
"source": "-1"
},
"108": {
"type": 1,
"opType": 3,
"lhs": "109",
"rhs": "110"
},
"109": {
"type": 5,
"name": "i31"
},
"110": {
"type": 0,
"val": 1536
},
"111": {
"type": 1,
"opType": 4,
"lhs": "112",
"rhs": "115"
},
"112": {
"type": 1,
"opType": 3,
"lhs": "113",
"rhs": "114"
},
"113": {
"type": 5,
"name": "i31"
},
"114": {
"type": 0,
"val": 512
},
"115": {
"type": 0,
"val": 3
},
"116": {
"type": 1,
"opType": 4,
"lhs": "117",
"rhs": "118"
},
"117": {
"type": 5,
"name": "i31"
},
"118": {
"type": 0,
"val": 512
},
"119": {
"type": 5,
"name": "c"
},
"120": {
"type": 2,
"paddings": [
0,
0
],
"summand": "121",
"loopVarRanges": {
"i32": [
0,
49
],
"i31": [
0,
4608
]
},
"sumVarRanges": {
"c": [
0,
512
]
}
},
"121": {
"type": 1,
"opType": 2,
"lhs": "122",
"rhs": "141"
},
"122": {
"type": 3,
"subExprsNum": 2,
"object": "123",
"indexes": [
"139",
"140"
]
},
"123": {
"type": 4,
"name": "T4",
"shape": [
49,
512
],
"paddings": [
0,
0
],
"source": "124"
},
"124": {
"type": 105,
"inputs": [
"125"
],
"expr": "126",
"outputShape": [
49,
512
]
},
"125": {
"type": 4,
"name": "A",
"shape": [
1,
7,
7,
512
],
"paddings": [
0,
1,
1,
0
],
"source": "-1"
},
"126": {
"type": 2,
"paddings": [
0,
0
],
"summand": "127",
"loopVarRanges": {
"i32": [
0,
49
],
"c": [
0,
512
]
}
},
"127": {
"type": 3,
"subExprsNum": 4,
"object": "128",
"indexes": [
"129",
"132",
"135",
"138"
]
},
"128": {
"type": 4,
"name": "A",
"shape": [
1,
7,
7,
512
],
"paddings": [
0,
1,
1,
0
],
"source": "-1"
},
"129": {
"type": 1,
"opType": 3,
"lhs": "130",
"rhs": "131"
},
"130": {
"type": 5,
"name": "i32"
},
"131": {
"type": 0,
"val": 49
},
"132": {
"type": 1,
"opType": 3,
"lhs": "133",
"rhs": "134"
},
"133": {
"type": 5,
"name": "i32"
},
"134": {
"type": 0,
"val": 7
},
"135": {
"type": 1,
"opType": 4,
"lhs": "136",
"rhs": "137"
},
"136": {
"type": 5,
"name": "i32"
},
"137": {
"type": 0,
"val": 7
},
"138": {
"type": 5,
"name": "c"
},
"139": {
"type": 5,
"name": "i32"
},
"140": {
"type": 5,
"name": "c"
},
"141": {
"type": 3,
"subExprsNum": 2,
"object": "142",
"indexes": [
"160",
"161"
]
},
"142": {
"type": 4,
"name": "T3",
"shape": [
4608,
512
],
"paddings": [
0,
0
],
"source": "143"
},
"143": {
"type": 105,
"inputs": [
"144"
],
"expr": "145",
"outputShape": [
4608,
512
]
},
"144": {
"type": 4,
"name": "K",
"shape": [
3,
3,
512,
512
],
"paddings": [
0,
0,
0,
0
],
"source": "-1"
},
"145": {
"type": 2,
"paddings": [
0,
0
],
"summand": "146",
"loopVarRanges": {
"i31": [
0,
4608
],
"c": [
0,
512
]
}
},
"146": {
"type": 3,
"subExprsNum": 4,
"object": "147",
"indexes": [
"148",
"151",
"156",
"159"
]
},
"147": {
"type": 4,
"name": "K",
"shape": [
3,
3,
512,
512
],
"paddings": [
0,
0,
0,
0
],
"source": "-1"
},
"148": {
"type": 1,
"opType": 3,
"lhs": "149",
"rhs": "150"
},
"149": {
"type": 5,
"name": "i31"
},
"150": {
"type": 0,
"val": 1536
},
"151": {
"type": 1,
"opType": 4,
"lhs": "152",
"rhs": "155"
},
"152": {
"type": 1,
"opType": 3,
"lhs": "153",
"rhs": "154"
},
"153": {
"type": 5,
"name": "i31"
},
"154": {
"type": 0,
"val": 512
},
"155": {
"type": 0,
"val": 3
},
"156": {
"type": 1,
"opType": 4,
"lhs": "157",
"rhs": "158"
},
"157": {
"type": 5,
"name": "i31"
},
"158": {
"type": 0,
"val": 512
},
"159": {
"type": 5,
"name": "c"
},
"160": {
"type": 5,
"name": "i31"
},
"161": {
"type": 5,
"name": "c"
},
"162": {
"type": 1,
"opType": 1,
"lhs": "163",
"rhs": "172"
},
"163": {
"type": 1,
"opType": 1,
"lhs": "164",
"rhs": "167"
},
"164": {
"type": 1,
"opType": 2,
"lhs": "165",
"rhs": "166"
},
"165": {
"type": 0,
"val": 49
},
"166": {
"type": 5,
"name": "n"
},
"167": {
"type": 1,
"opType": 2,
"lhs": "168",
"rhs": "169"
},
"168": {
"type": 0,
"val": 7
},
"169": {
"type": 1,
"opType": 1,
"lhs": "170",
"rhs": "171"
},
"170": {
"type": 5,
"name": "i17"
},
"171": {
"type": 0,
"val": -1
},
"172": {
"type": 1,
"opType": 1,
"lhs": "173",
"rhs": "174"
},
"173": {
"type": 5,
"name": "i27"
},
"174": {
"type": 0,
"val": -1
},
"175": {
"type": 1,
"opType": 1,
"lhs": "176",
"rhs": "183"
},
"176": {
"type": 1,
"opType": 1,
"lhs": "177",
"rhs": "180"
},
"177": {
"type": 1,
"opType": 2,
"lhs": "178",
"rhs": "179"
},
"178": {
"type": 0,
"val": 1536
},
"179": {
"type": 5,
"name": "i18"
},
"180": {
"type": 1,
"opType": 2,
"lhs": "181",
"rhs": "182"
},
"181": {
"type": 0,
"val": 512
},
"182": {
"type": 5,
"name": "i28"
},
"183": {
"type": 5,
"name": "f"
},
"184": {
"type": 1,
"opType": 1,
"lhs": "185",
"rhs": "186"
},
"185": {
"type": 5,
"name": "w"
},
"186": {
"type": 5,
"name": "s"
},
"187": {
"type": 5,
"name": "s"
},
"188": {
"type": 1,
"opType": 1,
"lhs": "189",
"rhs": "190"
},
"189": {
"type": 5,
"name": "h"
},
"190": {
"type": 5,
"name": "r"
},
"191": {
"type": 5,
"name": "r"
},
"192": {
"type": 5,
"name": "n"
},
"193": {
"type": 5,
"name": "f"
}
}