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

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