forked from jiuyuan/InfiniTensor
594 lines
9.3 KiB
Plaintext
594 lines
9.3 KiB
Plaintext
{
|
|
"Version": 1,
|
|
"0": {
|
|
"type": 2,
|
|
"paddings": [
|
|
0,
|
|
0,
|
|
0,
|
|
0
|
|
],
|
|
"summand": "1",
|
|
"loopVarRanges": {
|
|
"n": [
|
|
0,
|
|
1
|
|
],
|
|
"h": [
|
|
0,
|
|
7
|
|
],
|
|
"w": [
|
|
0,
|
|
7
|
|
],
|
|
"f": [
|
|
0,
|
|
512
|
|
]
|
|
},
|
|
"sumVarRanges": {
|
|
"r": [
|
|
0,
|
|
3
|
|
],
|
|
"s": [
|
|
0,
|
|
3
|
|
]
|
|
}
|
|
},
|
|
"1": {
|
|
"type": 3,
|
|
"subExprsNum": 6,
|
|
"object": "2",
|
|
"indexes": [
|
|
"68",
|
|
"71",
|
|
"72",
|
|
"75",
|
|
"76",
|
|
"77"
|
|
]
|
|
},
|
|
"2": {
|
|
"type": 2,
|
|
"paddings": [
|
|
1,
|
|
0,
|
|
1,
|
|
0,
|
|
0,
|
|
0
|
|
],
|
|
"summand": "3",
|
|
"loopVarRanges": {
|
|
"i27": [
|
|
1,
|
|
8
|
|
],
|
|
"i28": [
|
|
0,
|
|
3
|
|
],
|
|
"i17": [
|
|
1,
|
|
8
|
|
],
|
|
"i18": [
|
|
0,
|
|
3
|
|
],
|
|
"n": [
|
|
0,
|
|
1
|
|
],
|
|
"f": [
|
|
0,
|
|
512
|
|
]
|
|
}
|
|
},
|
|
"3": {
|
|
"type": 3,
|
|
"subExprsNum": 2,
|
|
"object": "4",
|
|
"indexes": [
|
|
"46",
|
|
"59"
|
|
]
|
|
},
|
|
"4": {
|
|
"type": 2,
|
|
"paddings": [
|
|
0,
|
|
0
|
|
],
|
|
"summand": "5",
|
|
"loopVarRanges": {
|
|
"i32": [
|
|
0,
|
|
49
|
|
],
|
|
"i31": [
|
|
0,
|
|
4608
|
|
]
|
|
},
|
|
"sumVarRanges": {
|
|
"c": [
|
|
0,
|
|
512
|
|
]
|
|
}
|
|
},
|
|
"5": {
|
|
"type": 1,
|
|
"opType": 2,
|
|
"lhs": "6",
|
|
"rhs": "25"
|
|
},
|
|
"6": {
|
|
"type": 3,
|
|
"subExprsNum": 2,
|
|
"object": "7",
|
|
"indexes": [
|
|
"23",
|
|
"24"
|
|
]
|
|
},
|
|
"7": {
|
|
"type": 4,
|
|
"name": "T4",
|
|
"shape": [
|
|
49,
|
|
512
|
|
],
|
|
"paddings": [
|
|
0,
|
|
0
|
|
],
|
|
"source": "8"
|
|
},
|
|
"8": {
|
|
"type": 105,
|
|
"inputs": [
|
|
"9"
|
|
],
|
|
"expr": "10",
|
|
"outputShape": [
|
|
49,
|
|
512
|
|
]
|
|
},
|
|
"9": {
|
|
"type": 4,
|
|
"name": "A",
|
|
"shape": [
|
|
1,
|
|
7,
|
|
7,
|
|
512
|
|
],
|
|
"paddings": [
|
|
0,
|
|
1,
|
|
1,
|
|
0
|
|
],
|
|
"source": "-1"
|
|
},
|
|
"10": {
|
|
"type": 2,
|
|
"paddings": [
|
|
0,
|
|
0
|
|
],
|
|
"summand": "11",
|
|
"loopVarRanges": {
|
|
"i32": [
|
|
0,
|
|
49
|
|
],
|
|
"c": [
|
|
0,
|
|
512
|
|
]
|
|
}
|
|
},
|
|
"11": {
|
|
"type": 3,
|
|
"subExprsNum": 4,
|
|
"object": "12",
|
|
"indexes": [
|
|
"13",
|
|
"16",
|
|
"19",
|
|
"22"
|
|
]
|
|
},
|
|
"12": {
|
|
"type": 4,
|
|
"name": "A",
|
|
"shape": [
|
|
1,
|
|
7,
|
|
7,
|
|
512
|
|
],
|
|
"paddings": [
|
|
0,
|
|
1,
|
|
1,
|
|
0
|
|
],
|
|
"source": "-1"
|
|
},
|
|
"13": {
|
|
"type": 1,
|
|
"opType": 3,
|
|
"lhs": "14",
|
|
"rhs": "15"
|
|
},
|
|
"14": {
|
|
"type": 5,
|
|
"name": "i32"
|
|
},
|
|
"15": {
|
|
"type": 0,
|
|
"val": 49
|
|
},
|
|
"16": {
|
|
"type": 1,
|
|
"opType": 3,
|
|
"lhs": "17",
|
|
"rhs": "18"
|
|
},
|
|
"17": {
|
|
"type": 5,
|
|
"name": "i32"
|
|
},
|
|
"18": {
|
|
"type": 0,
|
|
"val": 7
|
|
},
|
|
"19": {
|
|
"type": 1,
|
|
"opType": 4,
|
|
"lhs": "20",
|
|
"rhs": "21"
|
|
},
|
|
"20": {
|
|
"type": 5,
|
|
"name": "i32"
|
|
},
|
|
"21": {
|
|
"type": 0,
|
|
"val": 7
|
|
},
|
|
"22": {
|
|
"type": 5,
|
|
"name": "c"
|
|
},
|
|
"23": {
|
|
"type": 5,
|
|
"name": "i32"
|
|
},
|
|
"24": {
|
|
"type": 5,
|
|
"name": "c"
|
|
},
|
|
"25": {
|
|
"type": 3,
|
|
"subExprsNum": 2,
|
|
"object": "26",
|
|
"indexes": [
|
|
"44",
|
|
"45"
|
|
]
|
|
},
|
|
"26": {
|
|
"type": 4,
|
|
"name": "T3",
|
|
"shape": [
|
|
4608,
|
|
512
|
|
],
|
|
"paddings": [
|
|
0,
|
|
0
|
|
],
|
|
"source": "27"
|
|
},
|
|
"27": {
|
|
"type": 105,
|
|
"inputs": [
|
|
"28"
|
|
],
|
|
"expr": "29",
|
|
"outputShape": [
|
|
4608,
|
|
512
|
|
]
|
|
},
|
|
"28": {
|
|
"type": 4,
|
|
"name": "K",
|
|
"shape": [
|
|
3,
|
|
3,
|
|
512,
|
|
512
|
|
],
|
|
"paddings": [
|
|
0,
|
|
0,
|
|
0,
|
|
0
|
|
],
|
|
"source": "-1"
|
|
},
|
|
"29": {
|
|
"type": 2,
|
|
"paddings": [
|
|
0,
|
|
0
|
|
],
|
|
"summand": "30",
|
|
"loopVarRanges": {
|
|
"i31": [
|
|
0,
|
|
4608
|
|
],
|
|
"c": [
|
|
0,
|
|
512
|
|
]
|
|
}
|
|
},
|
|
"30": {
|
|
"type": 3,
|
|
"subExprsNum": 4,
|
|
"object": "31",
|
|
"indexes": [
|
|
"32",
|
|
"35",
|
|
"40",
|
|
"43"
|
|
]
|
|
},
|
|
"31": {
|
|
"type": 4,
|
|
"name": "K",
|
|
"shape": [
|
|
3,
|
|
3,
|
|
512,
|
|
512
|
|
],
|
|
"paddings": [
|
|
0,
|
|
0,
|
|
0,
|
|
0
|
|
],
|
|
"source": "-1"
|
|
},
|
|
"32": {
|
|
"type": 1,
|
|
"opType": 3,
|
|
"lhs": "33",
|
|
"rhs": "34"
|
|
},
|
|
"33": {
|
|
"type": 5,
|
|
"name": "i31"
|
|
},
|
|
"34": {
|
|
"type": 0,
|
|
"val": 1536
|
|
},
|
|
"35": {
|
|
"type": 1,
|
|
"opType": 4,
|
|
"lhs": "36",
|
|
"rhs": "39"
|
|
},
|
|
"36": {
|
|
"type": 1,
|
|
"opType": 3,
|
|
"lhs": "37",
|
|
"rhs": "38"
|
|
},
|
|
"37": {
|
|
"type": 5,
|
|
"name": "i31"
|
|
},
|
|
"38": {
|
|
"type": 0,
|
|
"val": 512
|
|
},
|
|
"39": {
|
|
"type": 0,
|
|
"val": 3
|
|
},
|
|
"40": {
|
|
"type": 1,
|
|
"opType": 4,
|
|
"lhs": "41",
|
|
"rhs": "42"
|
|
},
|
|
"41": {
|
|
"type": 5,
|
|
"name": "i31"
|
|
},
|
|
"42": {
|
|
"type": 0,
|
|
"val": 512
|
|
},
|
|
"43": {
|
|
"type": 5,
|
|
"name": "c"
|
|
},
|
|
"44": {
|
|
"type": 5,
|
|
"name": "i31"
|
|
},
|
|
"45": {
|
|
"type": 5,
|
|
"name": "c"
|
|
},
|
|
"46": {
|
|
"type": 1,
|
|
"opType": 1,
|
|
"lhs": "47",
|
|
"rhs": "56"
|
|
},
|
|
"47": {
|
|
"type": 1,
|
|
"opType": 1,
|
|
"lhs": "48",
|
|
"rhs": "51"
|
|
},
|
|
"48": {
|
|
"type": 1,
|
|
"opType": 2,
|
|
"lhs": "49",
|
|
"rhs": "50"
|
|
},
|
|
"49": {
|
|
"type": 0,
|
|
"val": 49
|
|
},
|
|
"50": {
|
|
"type": 5,
|
|
"name": "n"
|
|
},
|
|
"51": {
|
|
"type": 1,
|
|
"opType": 2,
|
|
"lhs": "52",
|
|
"rhs": "53"
|
|
},
|
|
"52": {
|
|
"type": 0,
|
|
"val": 7
|
|
},
|
|
"53": {
|
|
"type": 1,
|
|
"opType": 1,
|
|
"lhs": "54",
|
|
"rhs": "55"
|
|
},
|
|
"54": {
|
|
"type": 5,
|
|
"name": "i17"
|
|
},
|
|
"55": {
|
|
"type": 0,
|
|
"val": -1
|
|
},
|
|
"56": {
|
|
"type": 1,
|
|
"opType": 1,
|
|
"lhs": "57",
|
|
"rhs": "58"
|
|
},
|
|
"57": {
|
|
"type": 5,
|
|
"name": "i27"
|
|
},
|
|
"58": {
|
|
"type": 0,
|
|
"val": -1
|
|
},
|
|
"59": {
|
|
"type": 1,
|
|
"opType": 1,
|
|
"lhs": "60",
|
|
"rhs": "67"
|
|
},
|
|
"60": {
|
|
"type": 1,
|
|
"opType": 1,
|
|
"lhs": "61",
|
|
"rhs": "64"
|
|
},
|
|
"61": {
|
|
"type": 1,
|
|
"opType": 2,
|
|
"lhs": "62",
|
|
"rhs": "63"
|
|
},
|
|
"62": {
|
|
"type": 0,
|
|
"val": 1536
|
|
},
|
|
"63": {
|
|
"type": 5,
|
|
"name": "i18"
|
|
},
|
|
"64": {
|
|
"type": 1,
|
|
"opType": 2,
|
|
"lhs": "65",
|
|
"rhs": "66"
|
|
},
|
|
"65": {
|
|
"type": 0,
|
|
"val": 512
|
|
},
|
|
"66": {
|
|
"type": 5,
|
|
"name": "i28"
|
|
},
|
|
"67": {
|
|
"type": 5,
|
|
"name": "f"
|
|
},
|
|
"68": {
|
|
"type": 1,
|
|
"opType": 1,
|
|
"lhs": "69",
|
|
"rhs": "70"
|
|
},
|
|
"69": {
|
|
"type": 5,
|
|
"name": "w"
|
|
},
|
|
"70": {
|
|
"type": 5,
|
|
"name": "s"
|
|
},
|
|
"71": {
|
|
"type": 5,
|
|
"name": "s"
|
|
},
|
|
"72": {
|
|
"type": 1,
|
|
"opType": 1,
|
|
"lhs": "73",
|
|
"rhs": "74"
|
|
},
|
|
"73": {
|
|
"type": 5,
|
|
"name": "h"
|
|
},
|
|
"74": {
|
|
"type": 5,
|
|
"name": "r"
|
|
},
|
|
"75": {
|
|
"type": 5,
|
|
"name": "r"
|
|
},
|
|
"76": {
|
|
"type": 5,
|
|
"name": "n"
|
|
},
|
|
"77": {
|
|
"type": 5,
|
|
"name": "f"
|
|
}
|
|
}
|