stablehlo.broadcast_in_dim

STABLE HLO Input Variationsttnn opTorch NameStatus
0Tensor<[1,32,32,32]>,
dims: [0, 1, 2, 3]
aten::_safe_softmax4
1Tensor<[1,32,32,1]>,
dims: [0, 1, 2, 3]
aten::_safe_softmax4
2Scalar,
dims: []
aten::_safe_softmax4
3Tensor<[1,1,32,32]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
4Tensor<[1,1,1,32]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
5Tensor<[1,32,1]>,
dims: [0, 1, 2]
aten::add.Tensor4
6Tensor<[32]>,
dims: [0]
aten::arange4
7Tensor<[1,1,32]>,
dims: [0, 1, 2]
aten::bmm4
8Tensor<[32,128,32]>,
dims: [0, 1, 2]
aten::bmm4
9Tensor<[32,32,128]>,
dims: [0, 1, 2]
aten::bmm4
10Tensor<[32]>,
dims: [1]
aten::gt.Tensor4
11Tensor<[32,1]>,
dims: [0, 1]
aten::gt.Tensor4
12Tensor<[1,32,32,128]>,
dims: [0, 1, 2, 3]
aten::mul.Scalar4
13Tensor<[1,32,128,32]>,
dims: [0, 1, 2, 3]
aten::mul.Scalar4
14Tensor<[1,32,128]>,
dims: [0, 1, 2]
aten::mul.Tensor4
15Tensor<[1,32,4096]>,
dims: [0, 1, 2]
aten::mul.Tensor4
16Tensor<[4096]>,
dims: [2]
aten::mul.Tensor4
17Tensor<[1,1,32,128]>,
dims: [0, 1, 2, 3]
aten::mul.Tensor4
18Tensor<[1,32]>,
dims: [0, 1]
aten::triu4
19Tensor<[32,32]>,
dims: [0, 1]
aten::triu4
20Tensor<[1,12,7,7]>,
dims: [0, 1, 2, 3]
aten::_safe_softmax4
21Tensor<[1,12,7,1]>,
dims: [0, 1, 2, 3]
aten::_safe_softmax4
22Tensor<[7]>,
dims: [0]
aten::add.Tensor4
23Tensor<[1,7,1]>,
dims: [0, 1, 2]
aten::add.Tensor4
24Tensor<[1,7,768]>,
dims: [0, 1, 2]
aten::add.Tensor4
25Tensor<[768]>,
dims: [2]
aten::add.Tensor4
26Tensor<[7,2304]>,
dims: [0, 1]
aten::add.Tensor4
27Tensor<[2304]>,
dims: [1]
aten::add.Tensor4
28Tensor<[1,1,7,7]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
29Tensor<[7,768]>,
dims: [0, 1]
aten::add.Tensor4
30Tensor<[768]>,
dims: [1]
aten::add.Tensor4
31Tensor<[7,3072]>,
dims: [0, 1]
aten::add.Tensor4
32Tensor<[3072]>,
dims: [1]
aten::add.Tensor4
33Tensor<[1,7,3072]>,
dims: [0, 1, 2]
aten::add.Tensor4
34Tensor<[1]>,
dims: [0]
aten::arange4
35Tensor<[12,64,7]>,
dims: [0, 1, 2]
aten::bmm4
36Tensor<[12,7,64]>,
dims: [0, 1, 2]
aten::bmm4
37Tensor<[1,7]>,
dims: [0, 1]
aten::eq.Scalar4
38Tensor<[1,1,1,7]>,
dims: [0, 1, 2, 3]
aten::expand4
39Tensor<[7]>,
dims: [1]
aten::lt.Tensor4
40Tensor<[7,1]>,
dims: [0, 1]
aten::lt.Tensor4
41Tensor<[1,12,7,64]>,
dims: [0, 1, 2, 3]
aten::mul.Scalar4
42Tensor<[1,12,64,7]>,
dims: [0, 1, 2, 3]
aten::mul.Scalar4
43Tensor<[2304]>,
dims: [0]
aten::mul.Tensor4
44Tensor<[768]>,
dims: [0]
aten::mul.Tensor4
45Tensor<[3072]>,
dims: [0]
aten::mul.Tensor4
46Tensor<[7,7]>,
dims: [0, 1]
aten::where.self4
47Tensor<[1,32,112,112]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
48Tensor<[32,1,1]>,
dims: [1, 2, 3]
aten::add.Tensor4
49Tensor<[64]>,
dims: [0]
aten::add.Tensor4
50Tensor<[1,64,112,112]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
51Tensor<[64,1,1]>,
dims: [1, 2, 3]
aten::add.Tensor4
52Tensor<[1,64,56,56]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
53Tensor<[128]>,
dims: [0]
aten::add.Tensor4
54Tensor<[1,128,56,56]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
55Tensor<[128,1,1]>,
dims: [1, 2, 3]
aten::add.Tensor4
56Tensor<[1,128,28,28]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
57Tensor<[256]>,
dims: [0]
aten::add.Tensor4
58Tensor<[1,256,28,28]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
59Tensor<[256,1,1]>,
dims: [1, 2, 3]
aten::add.Tensor4
60Tensor<[512]>,
dims: [0]
aten::add.Tensor4
61Tensor<[1,512,28,28]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
62Tensor<[512,1,1]>,
dims: [1, 2, 3]
aten::add.Tensor4
63Tensor<[1,19,28,28]>,
dims: [0, 1, 2, 3]
aten::convolution4
64Tensor<[19,1,1]>,
dims: [1, 2, 3]
aten::convolution4
65Tensor<[1,38,28,28]>,
dims: [0, 1, 2, 3]
aten::convolution4
66Tensor<[38,1,1]>,
dims: [1, 2, 3]
aten::convolution4
67Tensor<[256,512]>,
dims: [0, 1]
aten::add.Tensor4
68Tensor<[512]>,
dims: [1]
aten::add.Tensor4
69Tensor<[1,256,1]>,
dims: [0, 1, 2]
aten::add.Tensor4
70Tensor<[1,256,512]>,
dims: [0, 1, 2]
aten::add.Tensor4
71Tensor<[512]>,
dims: [2]
aten::add.Tensor4
72Tensor<[256,256]>,
dims: [0, 1]
aten::add.Tensor4
73Tensor<[256]>,
dims: [1]
aten::add.Tensor4
74Tensor<[1,1000]>,
dims: [0, 1]
aten::add.Tensor4
75Tensor<[1000]>,
dims: [1]
aten::add.Tensor4
76Tensor<[1,1024,512]>,
dims: [0, 1, 2]
aten::convolution4
77Tensor<[1024,1]>,
dims: [1, 2]
aten::convolution4
78Tensor<[256,1]>,
dims: [1, 2]
aten::convolution4
79Tensor<[1,512]>,
dims: [0, 1]
aten::mean.dim4
80Tensor<[1000]>,
dims: [0]
aten::mul.Tensor4
81Tensor<[8,920,920]>,
dims: [0, 1, 2]
aten::_softmax4
82Tensor<[8,920,1]>,
dims: [0, 1, 2]
aten::_softmax4
83Tensor<[8,100,100]>,
dims: [0, 1, 2]
aten::_softmax4
84Tensor<[8,100,1]>,
dims: [0, 1, 2]
aten::_softmax4
85Tensor<[8,100,920]>,
dims: [0, 1, 2]
aten::_softmax4
86Tensor<[1,64,1,1]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
87Tensor<[1,64,360,640]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
88Tensor<[1,64,180,320]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
89Tensor<[1,256,1,1]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
90Tensor<[1,256,180,320]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
91Tensor<[1,128,1,1]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
92Tensor<[1,128,180,320]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
93Tensor<[1,128,90,160]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
94Tensor<[1,512,1,1]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
95Tensor<[1,512,90,160]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
96Tensor<[1,256,90,160]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
97Tensor<[1,256,45,80]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
98Tensor<[1,1024,1,1]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
99Tensor<[1,1024,45,80]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
100Tensor<[1,512,45,80]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
101Tensor<[1,512,23,40]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
102Tensor<[1,2048,1,1]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
103Tensor<[1,2048,23,40]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
104Tensor<[23]>,
dims: [0]
aten::add.Tensor4
105Tensor<[40]>,
dims: [0]
aten::add.Tensor4
106Tensor<[1,1,40]>,
dims: [0, 1, 2]
aten::add.Tensor4
107Tensor<[1,23,1]>,
dims: [0, 1, 2]
aten::add.Tensor4
108Tensor<[920,1,256]>,
dims: [0, 1, 2]
aten::add.Tensor4
109Tensor<[256]>,
dims: [2]
aten::add.Tensor4
110Tensor<[920,256]>,
dims: [0, 1]
aten::add.Tensor4
111Tensor<[920,1,1]>,
dims: [0, 1, 2]
aten::add.Tensor4
112Tensor<[920,2048]>,
dims: [0, 1]
aten::add.Tensor4
113Tensor<[2048]>,
dims: [1]
aten::add.Tensor4
114Tensor<[100,256]>,
dims: [0, 1]
aten::add.Tensor4
115Tensor<[100,1,1]>,
dims: [0, 1, 2]
aten::add.Tensor4
116Tensor<[100,1,256]>,
dims: [0, 1, 2]
aten::add.Tensor4
117Tensor<[100,2048]>,
dims: [0, 1]
aten::add.Tensor4
118Tensor<[6,1,100,92]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
119Tensor<[92]>,
dims: [3]
aten::add.Tensor4
120Tensor<[6,1,100,256]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
121Tensor<[256]>,
dims: [3]
aten::add.Tensor4
122Tensor<[6,1,100,4]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
123Tensor<[4]>,
dims: [3]
aten::add.Tensor4
124Tensor<[8,32,920]>,
dims: [0, 1, 2]
aten::baddbmm4
125Tensor<[8,1,920]>,
dims: [0, 1, 2]
aten::baddbmm4
126Tensor<[920,256,256]>,
dims: [0, 1, 2]
aten::bmm4
127Tensor<[8,920,32]>,
dims: [0, 1, 2]
aten::bmm4
128Tensor<[8,32,100]>,
dims: [0, 1, 2]
aten::bmm4
129Tensor<[8,100,32]>,
dims: [0, 1, 2]
aten::bmm4
130Tensor<[6,256,92]>,
dims: [0, 1, 2]
aten::bmm4
131Tensor<[6,256,256]>,
dims: [0, 1, 2]
aten::bmm4
132Tensor<[1,256,23,40]>,
dims: [0, 1, 2, 3]
aten::convolution4
133Tensor<[1,23,40]>,
dims: [0, 1, 2]
aten::div.Tensor4
134Tensor<[1,23,40,1]>,
dims: [0, 1, 2, 3]
aten::div.Tensor4
135Tensor<[128]>,
dims: [3]
aten::div.Tensor4
136Tensor<[256,256]>,
dims: [1, 2]
aten::expand5
137Tensor<[1,1,1,920]>,
dims: [0, 1, 2, 3]
aten::expand5
138Tensor<[256,92]>,
dims: [2, 3]
aten::expand5
139Tensor<[256,256]>,
dims: [2, 3]
aten::expand5
140Tensor<[1,1,1,1]>,
dims: [0, 1, 2, 3]
aten::index.Tensor4
141Tensor<[1,1,1]>,
dims: [1, 2, 3]
aten::index.Tensor4
142Tensor<[23,1]>,
dims: [2, 3]
aten::index.Tensor4
143Tensor<[40]>,
dims: [3]
aten::index.Tensor4
144Tensor<[2048]>,
dims: [0]
aten::mul.Tensor4
145Tensor<[1,920]>,
dims: [0, 1]
aten::where.self4
146Tensor<[1,12,10,10]>,
dims: [0, 1, 2, 3]
aten::_safe_softmax4
147Tensor<[1,12,10,1]>,
dims: [0, 1, 2, 3]
aten::_safe_softmax4
148Tensor<[1,10]>,
dims: [0, 1]
aten::add.Tensor5
149Tensor<[1,10,1]>,
dims: [0, 1, 2]
aten::add.Tensor4
150Tensor<[1,10,768]>,
dims: [0, 1, 2]
aten::add.Tensor4
151Tensor<[10,768]>,
dims: [0, 1]
aten::add.Tensor4
152Tensor<[1,1,10,10]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
153Tensor<[10,3072]>,
dims: [0, 1]
aten::add.Tensor4
154Tensor<[10,250002]>,
dims: [0, 1]
aten::add.Tensor4
155Tensor<[250002]>,
dims: [1]
aten::add.Tensor4
156Tensor<[12,64,10]>,
dims: [0, 1, 2]
aten::bmm4
157Tensor<[12,10,64]>,
dims: [0, 1, 2]
aten::bmm4
158Tensor<[1,1,1,10]>,
dims: [0, 1, 2, 3]
aten::expand4
159Tensor<[1,12,10,64]>,
dims: [0, 1, 2, 3]
aten::mul.Scalar4
160Tensor<[1,12,64,10]>,
dims: [0, 1, 2, 3]
aten::mul.Scalar4
161Tensor<[250002]>,
dims: [0]
aten::mul.Tensor4
162Tensor<[1,8,4096,4096]>,
dims: [0, 1, 2, 3]
aten::_safe_softmax4
163Tensor<[1,8,4096,1]>,
dims: [0, 1, 2, 3]
aten::_safe_softmax4
164Tensor<[1,8,4096,9]>,
dims: [0, 1, 2, 3]
aten::_safe_softmax4
165Tensor<[1,8,1024,1024]>,
dims: [0, 1, 2, 3]
aten::_safe_softmax4
166Tensor<[1,8,1024,1]>,
dims: [0, 1, 2, 3]
aten::_safe_softmax4
167Tensor<[1,8,1024,9]>,
dims: [0, 1, 2, 3]
aten::_safe_softmax4
168Tensor<[1,8,256,256]>,
dims: [0, 1, 2, 3]
aten::_safe_softmax4
169Tensor<[1,8,256,1]>,
dims: [0, 1, 2, 3]
aten::_safe_softmax4
170Tensor<[1,8,256,9]>,
dims: [0, 1, 2, 3]
aten::_safe_softmax4
171Tensor<[1,8,64,64]>,
dims: [0, 1, 2, 3]
aten::_safe_softmax4
172Tensor<[1,8,64,1]>,
dims: [0, 1, 2, 3]
aten::_safe_softmax4
173Tensor<[1,8,64,9]>,
dims: [0, 1, 2, 3]
aten::_safe_softmax4
174Tensor<[1,1280]>,
dims: [0, 1]
aten::add.Tensor4
175Tensor<[1280]>,
dims: [1]
aten::add.Tensor4
176Tensor<[1,32,1,1]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
177Tensor<[1,320,64,64]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
178Tensor<[1,320,1,1]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
179Tensor<[1,320]>,
dims: [0, 1]
aten::add.Tensor4
180Tensor<[320]>,
dims: [1]
aten::add.Tensor4
181Tensor<[1,4096,1]>,
dims: [0, 1, 2]
aten::add.Tensor4
182Tensor<[1,4096,320]>,
dims: [0, 1, 2]
aten::add.Tensor4
183Tensor<[320]>,
dims: [2]
aten::add.Tensor4
184Tensor<[4096,320]>,
dims: [0, 1]
aten::add.Tensor4
185Tensor<[4096,2560]>,
dims: [0, 1]
aten::add.Tensor4
186Tensor<[2560]>,
dims: [1]
aten::add.Tensor4
187Tensor<[1,320,32,32]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
188Tensor<[1,640]>,
dims: [0, 1]
aten::add.Tensor4
189Tensor<[640]>,
dims: [1]
aten::add.Tensor4
190Tensor<[1,640,32,32]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
191Tensor<[1,640,1,1]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
192Tensor<[1,1024,1]>,
dims: [0, 1, 2]
aten::add.Tensor4
193Tensor<[1,1024,640]>,
dims: [0, 1, 2]
aten::add.Tensor4
194Tensor<[640]>,
dims: [2]
aten::add.Tensor4
195Tensor<[1024,640]>,
dims: [0, 1]
aten::add.Tensor4
196Tensor<[1024,5120]>,
dims: [0, 1]
aten::add.Tensor4
197Tensor<[5120]>,
dims: [1]
aten::add.Tensor4
198Tensor<[1,640,16,16]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
199Tensor<[1,1280,16,16]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
200Tensor<[1,1280,1,1]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
201Tensor<[1,256,1280]>,
dims: [0, 1, 2]
aten::add.Tensor4
202Tensor<[1280]>,
dims: [2]
aten::add.Tensor4
203Tensor<[256,1280]>,
dims: [0, 1]
aten::add.Tensor4
204Tensor<[256,10240]>,
dims: [0, 1]
aten::add.Tensor4
205Tensor<[10240]>,
dims: [1]
aten::add.Tensor4
206Tensor<[1,1280,8,8]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
207Tensor<[1,64,1]>,
dims: [0, 1, 2]
aten::add.Tensor4
208Tensor<[1,64,1280]>,
dims: [0, 1, 2]
aten::add.Tensor4
209Tensor<[64,1280]>,
dims: [0, 1]
aten::add.Tensor4
210Tensor<[64,10240]>,
dims: [0, 1]
aten::add.Tensor4
211Tensor<[1,2560,8,8]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
212Tensor<[1,2560,1,1]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
213Tensor<[16]>,
dims: [0]
aten::add.Tensor4
214Tensor<[1,2560,16,16]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
215Tensor<[1,1920,16,16]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
216Tensor<[1,1920,1,1]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
217Tensor<[1,1920,32,32]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
218Tensor<[1,1280,32,32]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
219Tensor<[1,960,32,32]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
220Tensor<[1,960,1,1]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
221Tensor<[1,960,64,64]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
222Tensor<[1,640,64,64]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
223Tensor<[160]>,
dims: [0]
aten::arange.start4
224Tensor<[8,40,4096]>,
dims: [0, 1, 2]
aten::bmm4
225Tensor<[8,4096,40]>,
dims: [0, 1, 2]
aten::bmm4
226Tensor<[8,40,9]>,
dims: [0, 1, 2]
aten::bmm4
227Tensor<[8,9,40]>,
dims: [0, 1, 2]
aten::bmm4
228Tensor<[8,80,1024]>,
dims: [0, 1, 2]
aten::bmm4
229Tensor<[8,1024,80]>,
dims: [0, 1, 2]
aten::bmm4
230Tensor<[8,80,9]>,
dims: [0, 1, 2]
aten::bmm4
231Tensor<[8,9,80]>,
dims: [0, 1, 2]
aten::bmm4
232Tensor<[8,160,256]>,
dims: [0, 1, 2]
aten::bmm4
233Tensor<[8,256,160]>,
dims: [0, 1, 2]
aten::bmm4
234Tensor<[8,160,9]>,
dims: [0, 1, 2]
aten::bmm4
235Tensor<[8,9,160]>,
dims: [0, 1, 2]
aten::bmm4
236Tensor<[8,160,64]>,
dims: [0, 1, 2]
aten::bmm4
237Tensor<[8,64,160]>,
dims: [0, 1, 2]
aten::bmm4
238Tensor<[320,1,1]>,
dims: [1, 2, 3]
aten::convolution4
239Tensor<[640,1,1]>,
dims: [1, 2, 3]
aten::convolution4
240Tensor<[1280,1,1]>,
dims: [1, 2, 3]
aten::convolution4
241Tensor<[1,4,64,64]>,
dims: [0, 1, 2, 3]
aten::convolution4
242Tensor<[4,1,1]>,
dims: [1, 2, 3]
aten::convolution4
243Tensor<[1280]>,
dims: [0]
aten::index.Tensor4
244Tensor<[16,1]>,
dims: [2, 3]
aten::index.Tensor4
245Tensor<[16]>,
dims: [3]
aten::index.Tensor4
246Tensor<[32,1]>,
dims: [2, 3]
aten::index.Tensor4
247Tensor<[32]>,
dims: [3]
aten::index.Tensor4
248Tensor<[640]>,
dims: [0]
aten::index.Tensor4
249Tensor<[64,1]>,
dims: [2, 3]
aten::index.Tensor4
250Tensor<[64]>,
dims: [3]
aten::index.Tensor4
251Tensor<[1,8,4096,40]>,
dims: [0, 1, 2, 3]
aten::mul.Scalar4
252Tensor<[1,8,40,4096]>,
dims: [0, 1, 2, 3]
aten::mul.Scalar4
253Tensor<[1,8,40,9]>,
dims: [0, 1, 2, 3]
aten::mul.Scalar4
254Tensor<[1,8,1024,80]>,
dims: [0, 1, 2, 3]
aten::mul.Scalar4
255Tensor<[1,8,80,1024]>,
dims: [0, 1, 2, 3]
aten::mul.Scalar4
256Tensor<[1,8,80,9]>,
dims: [0, 1, 2, 3]
aten::mul.Scalar4
257Tensor<[1,8,256,160]>,
dims: [0, 1, 2, 3]
aten::mul.Scalar4
258Tensor<[1,8,160,256]>,
dims: [0, 1, 2, 3]
aten::mul.Scalar4
259Tensor<[1,8,160,9]>,
dims: [0, 1, 2, 3]
aten::mul.Scalar4
260Tensor<[1,8,64,160]>,
dims: [0, 1, 2, 3]
aten::mul.Scalar4
261Tensor<[1,8,160,64]>,
dims: [0, 1, 2, 3]
aten::mul.Scalar4
262Tensor<[1,1]>,
dims: [0, 1]
aten::mul.Tensor4
263Tensor<[1,160]>,
dims: [0, 1]
aten::mul.Tensor4
264Tensor<[1,32,10,4096]>,
dims: [0, 1, 2, 3]
aten::mul.Tensor4
265Tensor<[320]>,
dims: [0]
aten::mul.Tensor4
266Tensor<[2560]>,
dims: [0]
aten::mul.Tensor4
267Tensor<[1,32,10,1024]>,
dims: [0, 1, 2, 3]
aten::mul.Tensor4
268Tensor<[1,32,20,1024]>,
dims: [0, 1, 2, 3]
aten::mul.Tensor4
269Tensor<[5120]>,
dims: [0]
aten::mul.Tensor4
270Tensor<[1,32,20,256]>,
dims: [0, 1, 2, 3]
aten::mul.Tensor4
271Tensor<[1,32,40,256]>,
dims: [0, 1, 2, 3]
aten::mul.Tensor4
272Tensor<[10240]>,
dims: [0]
aten::mul.Tensor4
273Tensor<[1,32,40,64]>,
dims: [0, 1, 2, 3]
aten::mul.Tensor4
274Tensor<[1,32,80,64]>,
dims: [0, 1, 2, 3]
aten::mul.Tensor4
275Tensor<[1,32,80,256]>,
dims: [0, 1, 2, 3]
aten::mul.Tensor4
276Tensor<[1,32,60,256]>,
dims: [0, 1, 2, 3]
aten::mul.Tensor4
277Tensor<[1,32,60,1024]>,
dims: [0, 1, 2, 3]
aten::mul.Tensor4
278Tensor<[1,32,40,1024]>,
dims: [0, 1, 2, 3]
aten::mul.Tensor4
279Tensor<[1,32,30,1024]>,
dims: [0, 1, 2, 3]
aten::mul.Tensor4
280Tensor<[1,32,30,4096]>,
dims: [0, 1, 2, 3]
aten::mul.Tensor4
281Tensor<[1,32,20,4096]>,
dims: [0, 1, 2, 3]
aten::mul.Tensor4
282Tensor<[1,12,25,25]>,
dims: [0, 1, 2, 3]
aten::_safe_softmax4
283Tensor<[1,12,25,1]>,
dims: [0, 1, 2, 3]
aten::_safe_softmax4
284Tensor<[1,25,1]>,
dims: [0, 1, 2]
aten::add.Tensor4
285Tensor<[1,25,768]>,
dims: [0, 1, 2]
aten::add.Tensor4
286Tensor<[25,768]>,
dims: [0, 1]
aten::add.Tensor4
287Tensor<[1,1,25,25]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
288Tensor<[25,3072]>,
dims: [0, 1]
aten::add.Tensor4
289Tensor<[25,2]>,
dims: [0, 1]
aten::add.Tensor4
290Tensor<[2]>,
dims: [1]
aten::add.Tensor4
291Tensor<[1]>,
dims: [1]
aten::add.Tensor4
292Tensor<[12,64,25]>,
dims: [0, 1, 2]
aten::bmm4
293Tensor<[12,25,64]>,
dims: [0, 1, 2]
aten::bmm4
294Tensor<[1,1,1,25]>,
dims: [0, 1, 2, 3]
aten::expand4
295Tensor<[1,12,25,64]>,
dims: [0, 1, 2, 3]
aten::mul.Scalar4
296Tensor<[1,12,64,25]>,
dims: [0, 1, 2, 3]
aten::mul.Scalar4
297Tensor<[2]>,
dims: [0]
aten::mul.Tensor4
298Tensor<[1,3,1445,1445]>,
dims: [0, 1, 2, 3]
aten::_safe_softmax4
299Tensor<[1,3,1445,1]>,
dims: [0, 1, 2, 3]
aten::_safe_softmax4
300Tensor<[1,1445,1]>,
dims: [0, 1, 2]
aten::add.Tensor4
301Tensor<[1,1445,192]>,
dims: [0, 1, 2]
aten::add.Tensor4
302Tensor<[192]>,
dims: [2]
aten::add.Tensor4
303Tensor<[1445,192]>,
dims: [0, 1]
aten::add.Tensor4
304Tensor<[192]>,
dims: [1]
aten::add.Tensor4
305Tensor<[1445,768]>,
dims: [0, 1]
aten::add.Tensor4
306Tensor<[100,192]>,
dims: [0, 1]
aten::add.Tensor4
307Tensor<[100,92]>,
dims: [0, 1]
aten::add.Tensor4
308Tensor<[92]>,
dims: [1]
aten::add.Tensor4
309Tensor<[100,4]>,
dims: [0, 1]
aten::add.Tensor4
310Tensor<[4]>,
dims: [1]
aten::add.Tensor4
311Tensor<[3,64,1445]>,
dims: [0, 1, 2]
aten::bmm4
312Tensor<[3,1445,64]>,
dims: [0, 1, 2]
aten::bmm4
313Tensor<[1,192,32,42]>,
dims: [0, 1, 2, 3]
aten::convolution4
314Tensor<[192,1,1]>,
dims: [1, 2, 3]
aten::convolution4
315Tensor<[1,3,1445,64]>,
dims: [0, 1, 2, 3]
aten::mul.Scalar4
316Tensor<[1,3,64,1445]>,
dims: [0, 1, 2, 3]
aten::mul.Scalar4
317Tensor<[192]>,
dims: [0]
aten::mul.Tensor4
318Tensor<[92]>,
dims: [0]
aten::mul.Tensor4
319Tensor<[4]>,
dims: [0]
aten::mul.Tensor4
320Tensor<[1,256,14,14]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
321Tensor<[1,512,7,7]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
322Tensor<[1,12,8,8]>,
dims: [0, 1, 2, 3]
aten::_softmax4
323Tensor<[1,12,8,1]>,
dims: [0, 1, 2, 3]
aten::_softmax4
324Tensor<[1,8,1]>,
dims: [0, 1, 2]
aten::add.Tensor4
325Tensor<[1,8,768]>,
dims: [0, 1, 2]
aten::add.Tensor4
326Tensor<[1,1,1,8]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
327Tensor<[1,768]>,
dims: [0, 1]
aten::add.Tensor4
328Tensor<[1,3]>,
dims: [0, 1]
aten::add.Tensor4
329Tensor<[3]>,
dims: [1]
aten::add.Tensor4
330Tensor<[12,64,8]>,
dims: [0, 1, 2]
aten::bmm4
331Tensor<[12,8,64]>,
dims: [0, 1, 2]
aten::bmm4
332Tensor<[1,768,8]>,
dims: [0, 1, 2]
aten::convolution4
333Tensor<[768,1]>,
dims: [1, 2]
aten::convolution4
334Tensor<[1,3072,8]>,
dims: [0, 1, 2]
aten::convolution4
335Tensor<[3072,1]>,
dims: [1, 2]
aten::convolution4
336Tensor<[3]>,
dims: [0]
aten::mul.Tensor4
337Tensor<[1,8,256,2048]>,
dims: [0, 1, 2, 3]
aten::_softmax4
338Tensor<[1,8,2048,256]>,
dims: [0, 1, 2, 3]
aten::_softmax4
339Tensor<[1,8,2048,1]>,
dims: [0, 1, 2, 3]
aten::_softmax4
340Tensor<[1,2048,768]>,
dims: [0, 1, 2]
aten::add.Tensor4
341Tensor<[2048,768]>,
dims: [1, 2]
aten::add.Tensor4
342Tensor<[1,2048,1]>,
dims: [0, 1, 2]
aten::add.Tensor4
343Tensor<[2048,256]>,
dims: [0, 1]
aten::add.Tensor4
344Tensor<[2048,1280]>,
dims: [0, 1]
aten::add.Tensor4
345Tensor<[1,1,1,2048]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
346Tensor<[256,768]>,
dims: [0, 1]
aten::add.Tensor4
347Tensor<[2048,768]>,
dims: [0, 1]
aten::add.Tensor4
348Tensor<[2048,262]>,
dims: [0, 1]
aten::add.Tensor4
349Tensor<[262]>,
dims: [1]
aten::add.Tensor4
350Tensor<[8,32,2048]>,
dims: [0, 1, 2]
aten::bmm4
351Tensor<[8,2048,160]>,
dims: [0, 1, 2]
aten::bmm4
352Tensor<[8,32,256]>,
dims: [0, 1, 2]
aten::bmm4
353Tensor<[8,256,96]>,
dims: [0, 1, 2]
aten::bmm4
354Tensor<[256,1280]>,
dims: [1, 2]
aten::expand5
355Tensor<[1,256,56,56]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
356Tensor<[1024]>,
dims: [0]
aten::add.Tensor4
357Tensor<[1,1024,14,14]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
358Tensor<[1024,1,1]>,
dims: [1, 2, 3]
aten::add.Tensor4
359Tensor<[1,512,14,14]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
360Tensor<[1,2048,7,7]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
361Tensor<[2048,1,1]>,
dims: [1, 2, 3]
aten::add.Tensor4
362Tensor<[1,12,201,201]>,
dims: [0, 1, 2, 3]
aten::_softmax4
363Tensor<[1,12,201,1]>,
dims: [0, 1, 2, 3]
aten::_softmax4
364Tensor<[12]>,
dims: [0]
aten::add.Tensor4
365Tensor<[1,201,1]>,
dims: [0, 1, 2]
aten::add.Tensor4
366Tensor<[1,201,768]>,
dims: [0, 1, 2]
aten::add.Tensor4
367Tensor<[201,768]>,
dims: [0, 1]
aten::add.Tensor4
368Tensor<[1,1,1,201]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
369Tensor<[201,3072]>,
dims: [0, 1]
aten::add.Tensor4
370Tensor<[1,1536]>,
dims: [0, 1]
aten::add.Tensor4
371Tensor<[1536]>,
dims: [1]
aten::add.Tensor4
372Tensor<[1,3129]>,
dims: [0, 1]
aten::add.Tensor4
373Tensor<[3129]>,
dims: [1]
aten::add.Tensor4
374Tensor<[12,64,201]>,
dims: [0, 1, 2]
aten::bmm4
375Tensor<[12,201,64]>,
dims: [0, 1, 2]
aten::bmm4
376Tensor<[1,768,12,16]>,
dims: [0, 1, 2, 3]
aten::convolution4
377Tensor<[768,1,1]>,
dims: [1, 2, 3]
aten::convolution4
378Tensor<[12,1]>,
dims: [0, 1]
aten::expand4
379Tensor<[1,16]>,
dims: [0, 1]
aten::expand4
380Tensor<[12,1]>,
dims: [2, 3]
aten::index.Tensor4
381Tensor<[1536]>,
dims: [0]
aten::mul.Tensor4
382Tensor<[3129]>,
dims: [0]
aten::mul.Tensor4
383Tensor<[1,192]>,
dims: [0, 1]
aten::rsub.Scalar4
384Tensor<[1,128]>,
dims: [0, 1]
aten::add.Tensor4
385Tensor<[128]>,
dims: [1]
aten::add.Tensor4
386Tensor<[10]>,
dims: [1]
aten::add.Tensor4
387Tensor<[1,32,26,26]>,
dims: [0, 1, 2, 3]
aten::convolution4
388Tensor<[1,64,24,24]>,
dims: [0, 1, 2, 3]
aten::convolution4
389Tensor<[10]>,
dims: [0]
aten::mul.Tensor4
390Tensor<[16,19,19]>,
dims: [0, 1, 2]
aten::_softmax4
391Tensor<[16,19,1]>,
dims: [0, 1, 2]
aten::_softmax4
392Tensor<[19]>,
dims: [0]
aten::add.Tensor4
393Tensor<[1,19]>,
dims: [0, 1]
aten::add.Tensor4
394Tensor<[1,19,1]>,
dims: [0, 1, 2]
aten::add.Tensor4
395Tensor<[1,19,1024]>,
dims: [0, 1, 2]
aten::add.Tensor4
396Tensor<[1024]>,
dims: [2]
aten::add.Tensor4
397Tensor<[19,1024]>,
dims: [0, 1]
aten::add.Tensor4
398Tensor<[1024]>,
dims: [1]
aten::add.Tensor4
399Tensor<[1,16,19,19]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
400Tensor<[1,1,19,19]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
401Tensor<[19,4096]>,
dims: [0, 1]
aten::add.Tensor4
402Tensor<[4096]>,
dims: [1]
aten::add.Tensor4
403Tensor<[16,64,19]>,
dims: [0, 1, 2]
aten::bmm4
404Tensor<[16,19,64]>,
dims: [0, 1, 2]
aten::bmm4
405Tensor<[1,1,1,19]>,
dims: [0, 1, 2, 3]
aten::expand4
406Tensor<[19]>,
dims: [1]
aten::lt.Tensor4
407Tensor<[19,1]>,
dims: [0, 1]
aten::lt.Tensor4
408Tensor<[4096]>,
dims: [0]
aten::mul.Tensor4
409Tensor<[19,256008]>,
dims: [0, 1]
aten::sub.Tensor4
410Tensor<[19,19]>,
dims: [0, 1]
aten::where.self4
411Tensor<[14]>,
dims: [0]
aten::add.Tensor4
412Tensor<[1,14,56,56]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
413Tensor<[14,1,1]>,
dims: [1, 2, 3]
aten::add.Tensor4
414Tensor<[24]>,
dims: [0]
aten::add.Tensor4
415Tensor<[1,24,56,56]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
416Tensor<[24,1,1]>,
dims: [1, 2, 3]
aten::add.Tensor4
417Tensor<[1,40,56,56]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
418Tensor<[40,1,1]>,
dims: [1, 2, 3]
aten::add.Tensor4
419Tensor<[68]>,
dims: [0]
aten::add.Tensor4
420Tensor<[1,68,56,56]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
421Tensor<[68,1,1]>,
dims: [1, 2, 3]
aten::add.Tensor4
422Tensor<[1,16,28,28]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
423Tensor<[16,1,1]>,
dims: [1, 2, 3]
aten::add.Tensor4
424Tensor<[28]>,
dims: [0]
aten::add.Tensor4
425Tensor<[1,28,28,28]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
426Tensor<[28,1,1]>,
dims: [1, 2, 3]
aten::add.Tensor4
427Tensor<[46]>,
dims: [0]
aten::add.Tensor4
428Tensor<[1,46,28,28]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
429Tensor<[46,1,1]>,
dims: [1, 2, 3]
aten::add.Tensor4
430Tensor<[78]>,
dims: [0]
aten::add.Tensor4
431Tensor<[1,78,28,28]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
432Tensor<[78,1,1]>,
dims: [1, 2, 3]
aten::add.Tensor4
433Tensor<[134]>,
dims: [0]
aten::add.Tensor4
434Tensor<[1,134,28,28]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
435Tensor<[134,1,1]>,
dims: [1, 2, 3]
aten::add.Tensor4
436Tensor<[20]>,
dims: [0]
aten::add.Tensor4
437Tensor<[1,20,28,28]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
438Tensor<[20,1,1]>,
dims: [1, 2, 3]
aten::add.Tensor4
439Tensor<[34]>,
dims: [0]
aten::add.Tensor4
440Tensor<[1,34,28,28]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
441Tensor<[34,1,1]>,
dims: [1, 2, 3]
aten::add.Tensor4
442Tensor<[58]>,
dims: [0]
aten::add.Tensor4
443Tensor<[1,58,28,28]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
444Tensor<[58,1,1]>,
dims: [1, 2, 3]
aten::add.Tensor4
445Tensor<[98]>,
dims: [0]
aten::add.Tensor4
446Tensor<[1,98,28,28]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
447Tensor<[98,1,1]>,
dims: [1, 2, 3]
aten::add.Tensor4
448Tensor<[168]>,
dims: [0]
aten::add.Tensor4
449Tensor<[1,168,28,28]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
450Tensor<[168,1,1]>,
dims: [1, 2, 3]
aten::add.Tensor4
451Tensor<[1,320,28,28]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
452Tensor<[1,40,14,14]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
453Tensor<[1,68,14,14]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
454Tensor<[116]>,
dims: [0]
aten::add.Tensor4
455Tensor<[1,116,14,14]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
456Tensor<[116,1,1]>,
dims: [1, 2, 3]
aten::add.Tensor4
457Tensor<[196]>,
dims: [0]
aten::add.Tensor4
458Tensor<[1,196,14,14]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
459Tensor<[196,1,1]>,
dims: [1, 2, 3]
aten::add.Tensor4
460Tensor<[334]>,
dims: [0]
aten::add.Tensor4
461Tensor<[1,334,14,14]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
462Tensor<[334,1,1]>,
dims: [1, 2, 3]
aten::add.Tensor4
463Tensor<[1,640,14,14]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
464Tensor<[1,160,7,7]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
465Tensor<[160,1,1]>,
dims: [1, 2, 3]
aten::add.Tensor4
466Tensor<[272]>,
dims: [0]
aten::add.Tensor4
467Tensor<[1,272,7,7]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
468Tensor<[272,1,1]>,
dims: [1, 2, 3]
aten::add.Tensor4
469Tensor<[462]>,
dims: [0]
aten::add.Tensor4
470Tensor<[1,462,7,7]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
471Tensor<[462,1,1]>,
dims: [1, 2, 3]
aten::add.Tensor4
472Tensor<[1,1024,7,7]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
473Tensor<[1,32,512,512]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
474Tensor<[1,64,256,256]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
475Tensor<[1,32,256,256]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
476Tensor<[1,128,128,128]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
477Tensor<[1,64,128,128]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
478Tensor<[1,256,64,64]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
479Tensor<[1,128,64,64]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
480Tensor<[1,512,32,32]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
481Tensor<[1,256,32,32]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
482Tensor<[1,1024,16,16]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
483Tensor<[1,512,16,16]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
484Tensor<[1,256,16,16]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
485Tensor<[1,128,32,32]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
486Tensor<[1,255,16,16]>,
dims: [0, 1, 2, 3]
aten::convolution4
487Tensor<[255,1,1]>,
dims: [1, 2, 3]
aten::convolution4
488Tensor<[1,255,32,32]>,
dims: [0, 1, 2, 3]
aten::convolution4
489Tensor<[1,255,64,64]>,
dims: [0, 1, 2, 3]
aten::convolution4
490Tensor<[1,1,256,256]>,
dims: [0, 1, 2, 3]
aten::convolution4
491Tensor<[1,4,14,14]>,
dims: [0, 1, 2, 3]
aten::convolution4
492Tensor<[1,16,14,14]>,
dims: [0, 1, 2, 3]
aten::convolution4
493Tensor<[1,1,28,28]>,
dims: [0, 1, 2, 3]
aten::convolution4
494Tensor<[1,16,32,32]>,
dims: [0, 1, 2, 3]
aten::_softmax4
495Tensor<[1,16,32,1]>,
dims: [0, 1, 2, 3]
aten::_softmax4
496Tensor<[1,32,1536]>,
dims: [0, 1, 2]
aten::add.Tensor4
497Tensor<[1536]>,
dims: [2]
aten::add.Tensor4
498Tensor<[32,4608]>,
dims: [0, 1]
aten::add.Tensor4
499Tensor<[4608]>,
dims: [1]
aten::add.Tensor4
500Tensor<[32,1536]>,
dims: [0, 1]
aten::add.Tensor4
501Tensor<[32,6144]>,
dims: [0, 1]
aten::add.Tensor4
502Tensor<[6144]>,
dims: [1]
aten::add.Tensor4
503Tensor<[1,32,6144]>,
dims: [0, 1, 2]
aten::add.Tensor4
504Tensor<[16,96,32]>,
dims: [0, 1, 2]
aten::baddbmm4
505Tensor<[16,32,32]>,
dims: [0, 1, 2]
aten::baddbmm4
506Tensor<[16,1,32]>,
dims: [0, 1, 2]
aten::baddbmm4
507Tensor<[16,32,96]>,
dims: [0, 1, 2]
aten::bmm4
508Tensor<[16,1]>,
dims: [1, 2]
aten::mul.Tensor4
509Tensor<[4608]>,
dims: [0]
aten::mul.Tensor4
510Tensor<[6144]>,
dims: [0]
aten::mul.Tensor4
511Tensor<[1,12,16,16]>,
dims: [0, 1, 2, 3]
aten::_safe_softmax4
512Tensor<[1,12,16,1]>,
dims: [0, 1, 2, 3]
aten::_safe_softmax4
513Tensor<[1,16,1]>,
dims: [0, 1, 2]
aten::add.Tensor4
514Tensor<[1,16,768]>,
dims: [0, 1, 2]
aten::add.Tensor4
515Tensor<[16,768]>,
dims: [0, 1]
aten::add.Tensor4
516Tensor<[1,1,16,16]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
517Tensor<[16,3072]>,
dims: [0, 1]
aten::add.Tensor4
518Tensor<[12,64,16]>,
dims: [0, 1, 2]
aten::bmm4
519Tensor<[12,16,64]>,
dims: [0, 1, 2]
aten::bmm4
520Tensor<[1,1,1,16]>,
dims: [0, 1, 2, 3]
aten::expand4
521Tensor<[1,12,16,64]>,
dims: [0, 1, 2, 3]
aten::mul.Scalar4
522Tensor<[1,12,64,16]>,
dims: [0, 1, 2, 3]
aten::mul.Scalar4
523Tensor<[1,64,224,224]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
524Tensor<[1,128,112,112]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
525Tensor<[1,1,224,224]>,
dims: [0, 1, 2, 3]
aten::convolution4
526Tensor<[1,1,19200,300]>,
dims: [0, 1, 2, 3]
aten::_softmax4
527Tensor<[1,1,19200,1]>,
dims: [0, 1, 2, 3]
aten::_softmax4
528Tensor<[1,2,4800,300]>,
dims: [0, 1, 2, 3]
aten::_softmax4
529Tensor<[1,2,4800,1]>,
dims: [0, 1, 2, 3]
aten::_softmax4
530Tensor<[1,5,1200,300]>,
dims: [0, 1, 2, 3]
aten::_softmax4
531Tensor<[1,5,1200,1]>,
dims: [0, 1, 2, 3]
aten::_softmax4
532Tensor<[1,8,300,300]>,
dims: [0, 1, 2, 3]
aten::_softmax4
533Tensor<[1,8,300,1]>,
dims: [0, 1, 2, 3]
aten::_softmax4
534Tensor<[1,19200,1]>,
dims: [0, 1, 2]
aten::add.Tensor4
535Tensor<[1,19200,64]>,
dims: [0, 1, 2]
aten::add.Tensor4
536Tensor<[64]>,
dims: [2]
aten::add.Tensor4
537Tensor<[19200,64]>,
dims: [0, 1]
aten::add.Tensor4
538Tensor<[64]>,
dims: [1]
aten::add.Tensor4
539Tensor<[1,300,1]>,
dims: [0, 1, 2]
aten::add.Tensor4
540Tensor<[1,300,64]>,
dims: [0, 1, 2]
aten::add.Tensor4
541Tensor<[300,64]>,
dims: [0, 1]
aten::add.Tensor4
542Tensor<[19200,256]>,
dims: [0, 1]
aten::add.Tensor4
543Tensor<[1,4800,1]>,
dims: [0, 1, 2]
aten::add.Tensor4
544Tensor<[1,4800,128]>,
dims: [0, 1, 2]
aten::add.Tensor4
545Tensor<[128]>,
dims: [2]
aten::add.Tensor4
546Tensor<[4800,128]>,
dims: [0, 1]
aten::add.Tensor4
547Tensor<[1,300,128]>,
dims: [0, 1, 2]
aten::add.Tensor4
548Tensor<[300,128]>,
dims: [0, 1]
aten::add.Tensor4
549Tensor<[4800,512]>,
dims: [0, 1]
aten::add.Tensor4
550Tensor<[1,1200,1]>,
dims: [0, 1, 2]
aten::add.Tensor4
551Tensor<[1,1200,320]>,
dims: [0, 1, 2]
aten::add.Tensor4
552Tensor<[1200,320]>,
dims: [0, 1]
aten::add.Tensor4
553Tensor<[1,300,320]>,
dims: [0, 1, 2]
aten::add.Tensor4
554Tensor<[300,320]>,
dims: [0, 1]
aten::add.Tensor4
555Tensor<[1200,1280]>,
dims: [0, 1]
aten::add.Tensor4
556Tensor<[1,300,512]>,
dims: [0, 1, 2]
aten::add.Tensor4
557Tensor<[300,512]>,
dims: [0, 1]
aten::add.Tensor4
558Tensor<[300,2048]>,
dims: [0, 1]
aten::add.Tensor4
559Tensor<[30]>,
dims: [0]
aten::add.Tensor4
560Tensor<[30,1]>,
dims: [0, 1]
aten::add.Tensor4
561Tensor<[1,64,30,40]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
562Tensor<[1,32,30,40]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
563Tensor<[60]>,
dims: [0]
aten::add.Tensor4
564Tensor<[60,1]>,
dims: [0, 1]
aten::add.Tensor4
565Tensor<[80]>,
dims: [0]
aten::add.Tensor4
566Tensor<[1,64,60,80]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
567Tensor<[1,32,60,80]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
568Tensor<[120]>,
dims: [0]
aten::add.Tensor4
569Tensor<[120,1]>,
dims: [0, 1]
aten::add.Tensor4
570Tensor<[1,64,120,160]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
571Tensor<[1,32,120,160]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
572Tensor<[240]>,
dims: [0]
aten::add.Tensor4
573Tensor<[240,1]>,
dims: [0, 1]
aten::add.Tensor4
574Tensor<[480]>,
dims: [0]
aten::add.Tensor4
575Tensor<[480,1]>,
dims: [0, 1]
aten::add.Tensor4
576Tensor<[1,64,300]>,
dims: [0, 1, 2]
aten::bmm4
577Tensor<[1,256,64]>,
dims: [0, 1, 2]
aten::bmm4
578Tensor<[2,64,300]>,
dims: [0, 1, 2]
aten::bmm4
579Tensor<[2,300,64]>,
dims: [0, 1, 2]
aten::bmm4
580Tensor<[1,512,128]>,
dims: [0, 1, 2]
aten::bmm4
581Tensor<[5,64,300]>,
dims: [0, 1, 2]
aten::bmm4
582Tensor<[5,300,64]>,
dims: [0, 1, 2]
aten::bmm4
583Tensor<[1,1280,320]>,
dims: [0, 1, 2]
aten::bmm4
584Tensor<[8,64,300]>,
dims: [0, 1, 2]
aten::bmm4
585Tensor<[8,300,64]>,
dims: [0, 1, 2]
aten::bmm4
586Tensor<[1,2048,512]>,
dims: [0, 1, 2]
aten::bmm4
587Tensor<[1,64,15,20]>,
dims: [0, 1, 2, 3]
aten::convolution4
588Tensor<[1,256,120,160]>,
dims: [0, 1, 2, 3]
aten::convolution4
589Tensor<[1,128,60,80]>,
dims: [0, 1, 2, 3]
aten::convolution4
590Tensor<[1,128,15,20]>,
dims: [0, 1, 2, 3]
aten::convolution4
591Tensor<[1,512,60,80]>,
dims: [0, 1, 2, 3]
aten::convolution4
592Tensor<[1,320,30,40]>,
dims: [0, 1, 2, 3]
aten::convolution4
593Tensor<[1,320,15,20]>,
dims: [0, 1, 2, 3]
aten::convolution4
594Tensor<[1,1280,30,40]>,
dims: [0, 1, 2, 3]
aten::convolution4
595Tensor<[1,512,15,20]>,
dims: [0, 1, 2, 3]
aten::convolution4
596Tensor<[1,2048,15,20]>,
dims: [0, 1, 2, 3]
aten::convolution4
597Tensor<[1,2,30,40]>,
dims: [0, 1, 2, 3]
aten::convolution4
598Tensor<[2,1,1]>,
dims: [1, 2, 3]
aten::convolution4
599Tensor<[1,2,60,80]>,
dims: [0, 1, 2, 3]
aten::convolution4
600Tensor<[1,2,120,160]>,
dims: [0, 1, 2, 3]
aten::convolution4
601Tensor<[1,64,480,640]>,
dims: [0, 1, 2, 3]
aten::convolution4
602Tensor<[1,1,480,640]>,
dims: [0, 1, 2, 3]
aten::convolution4
603Tensor<[256,64]>,
dims: [1, 2]
aten::expand5
604Tensor<[512,128]>,
dims: [1, 2]
aten::expand5
605Tensor<[1280,320]>,
dims: [1, 2]
aten::expand5
606Tensor<[2048,512]>,
dims: [1, 2]
aten::expand5
607Tensor<[30,1]>,
dims: [2, 3]
aten::index.Tensor4
608Tensor<[60,1]>,
dims: [2, 3]
aten::index.Tensor4
609Tensor<[80]>,
dims: [3]
aten::index.Tensor4
610Tensor<[120,1]>,
dims: [2, 3]
aten::index.Tensor4
611Tensor<[160]>,
dims: [3]
aten::index.Tensor4
612Tensor<[240,1]>,
dims: [2, 3]
aten::index.Tensor4
613Tensor<[320]>,
dims: [3]
aten::index.Tensor4
614Tensor<[480,1]>,
dims: [2, 3]
aten::index.Tensor4
615Tensor<[640]>,
dims: [3]
aten::index.Tensor4
616Tensor<[1,1,30,40]>,
dims: [0, 1, 2, 3]
aten::mul.Tensor4
617Tensor<[1,1,60,80]>,
dims: [0, 1, 2, 3]
aten::mul.Tensor4
618Tensor<[1,1,120,160]>,
dims: [0, 1, 2, 3]
aten::mul.Tensor4
619Tensor<[1,64,240,320]>,
dims: [0, 1, 2, 3]
aten::mul.Tensor4
620Tensor<[1,12,197,197]>,
dims: [0, 1, 2, 3]
aten::_safe_softmax4
621Tensor<[1,12,197,1]>,
dims: [0, 1, 2, 3]
aten::_safe_softmax4
622Tensor<[1,197,1]>,
dims: [0, 1, 2]
aten::add.Tensor4
623Tensor<[1,197,768]>,
dims: [0, 1, 2]
aten::add.Tensor4
624Tensor<[197,768]>,
dims: [0, 1]
aten::add.Tensor4
625Tensor<[197,3072]>,
dims: [0, 1]
aten::add.Tensor4
626Tensor<[12,64,197]>,
dims: [0, 1, 2]
aten::bmm4
627Tensor<[12,197,64]>,
dims: [0, 1, 2]
aten::bmm4
628Tensor<[1,768,14,14]>,
dims: [0, 1, 2, 3]
aten::convolution4
629Tensor<[1,12,197,64]>,
dims: [0, 1, 2, 3]
aten::mul.Scalar4
630Tensor<[1,12,64,197]>,
dims: [0, 1, 2, 3]
aten::mul.Scalar4
631Tensor<[1,1,16384,256]>,
dims: [0, 1, 2, 3]
aten::_softmax4
632Tensor<[1,1,16384,1]>,
dims: [0, 1, 2, 3]
aten::_softmax4
633Tensor<[1,2,4096,256]>,
dims: [0, 1, 2, 3]
aten::_softmax4
634Tensor<[1,2,4096,1]>,
dims: [0, 1, 2, 3]
aten::_softmax4
635Tensor<[1,5,1024,256]>,
dims: [0, 1, 2, 3]
aten::_softmax4
636Tensor<[1,5,1024,1]>,
dims: [0, 1, 2, 3]
aten::_softmax4
637Tensor<[1,16384,1]>,
dims: [0, 1, 2]
aten::add.Tensor4
638Tensor<[1,16384,32]>,
dims: [0, 1, 2]
aten::add.Tensor4
639Tensor<[32]>,
dims: [2]
aten::add.Tensor4
640Tensor<[16384,32]>,
dims: [0, 1]
aten::add.Tensor4
641Tensor<[1,256,32]>,
dims: [0, 1, 2]
aten::add.Tensor4
642Tensor<[256,32]>,
dims: [0, 1]
aten::add.Tensor4
643Tensor<[16384,128]>,
dims: [0, 1]
aten::add.Tensor4
644Tensor<[1,4096,64]>,
dims: [0, 1, 2]
aten::add.Tensor4
645Tensor<[4096,64]>,
dims: [0, 1]
aten::add.Tensor4
646Tensor<[256,64]>,
dims: [0, 1]
aten::add.Tensor4
647Tensor<[4096,256]>,
dims: [0, 1]
aten::add.Tensor4
648Tensor<[1,1024,160]>,
dims: [0, 1, 2]
aten::add.Tensor4
649Tensor<[160]>,
dims: [2]
aten::add.Tensor4
650Tensor<[1024,160]>,
dims: [0, 1]
aten::add.Tensor4
651Tensor<[160]>,
dims: [1]
aten::add.Tensor4
652Tensor<[1,256,160]>,
dims: [0, 1, 2]
aten::add.Tensor4
653Tensor<[256,160]>,
dims: [0, 1]
aten::add.Tensor4
654Tensor<[1,256,256]>,
dims: [0, 1, 2]
aten::add.Tensor4
655Tensor<[256,1024]>,
dims: [0, 1]
aten::add.Tensor4
656Tensor<[1,16384,256]>,
dims: [0, 1, 2]
aten::add.Tensor4
657Tensor<[128,1]>,
dims: [0, 1]
aten::add.Tensor4
658Tensor<[1,4096,256]>,
dims: [0, 1, 2]
aten::add.Tensor4
659Tensor<[1,1024,256]>,
dims: [0, 1, 2]
aten::add.Tensor4
660Tensor<[1,256,128,128]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
661Tensor<[1,32,256]>,
dims: [0, 1, 2]
aten::bmm4
662Tensor<[1,128,32]>,
dims: [0, 1, 2]
aten::bmm4
663Tensor<[2,32,256]>,
dims: [0, 1, 2]
aten::bmm4
664Tensor<[2,256,32]>,
dims: [0, 1, 2]
aten::bmm4
665Tensor<[5,32,256]>,
dims: [0, 1, 2]
aten::bmm4
666Tensor<[5,256,32]>,
dims: [0, 1, 2]
aten::bmm4
667Tensor<[1,640,160]>,
dims: [0, 1, 2]
aten::bmm4
668Tensor<[8,256,32]>,
dims: [0, 1, 2]
aten::bmm4
669Tensor<[1,64,256]>,
dims: [0, 1, 2]
aten::bmm4
670Tensor<[1,160,256]>,
dims: [0, 1, 2]
aten::bmm4
671Tensor<[1,32,128,128]>,
dims: [0, 1, 2, 3]
aten::convolution4
672Tensor<[1,32,16,16]>,
dims: [0, 1, 2, 3]
aten::convolution4
673Tensor<[1,64,64,64]>,
dims: [0, 1, 2, 3]
aten::convolution4
674Tensor<[1,64,16,16]>,
dims: [0, 1, 2, 3]
aten::convolution4
675Tensor<[1,160,32,32]>,
dims: [0, 1, 2, 3]
aten::convolution4
676Tensor<[1,160,16,16]>,
dims: [0, 1, 2, 3]
aten::convolution4
677Tensor<[1,150,128,128]>,
dims: [0, 1, 2, 3]
aten::convolution4
678Tensor<[150,1,1]>,
dims: [1, 2, 3]
aten::convolution4
679Tensor<[128,32]>,
dims: [1, 2]
aten::expand5
680Tensor<[640,160]>,
dims: [1, 2]
aten::expand5
681Tensor<[1024,256]>,
dims: [1, 2]
aten::expand5
682Tensor<[32,256]>,
dims: [1, 2]
aten::expand5
683Tensor<[64,256]>,
dims: [1, 2]
aten::expand5
684Tensor<[160,256]>,
dims: [1, 2]
aten::expand5
685Tensor<[128,1]>,
dims: [2, 3]
aten::index.Tensor4
686Tensor<[1,71,7,7]>,
dims: [0, 1, 2, 3]
aten::_safe_softmax4
687Tensor<[1,71,7,1]>,
dims: [0, 1, 2, 3]
aten::_safe_softmax4
688Tensor<[1,7,4544]>,
dims: [0, 1, 2]
aten::add.Tensor4
689Tensor<[4544]>,
dims: [2]
aten::add.Tensor4
690Tensor<[1,1,7]>,
dims: [0, 1, 2]
aten::bmm4
691Tensor<[71,64,7]>,
dims: [0, 1, 2]
aten::bmm4
692Tensor<[71,7,64]>,
dims: [0, 1, 2]
aten::bmm4
693Tensor<[1,1,64,7]>,
dims: [0, 1, 2, 3]
aten::expand5
694Tensor<[1,1,7,64]>,
dims: [0, 1, 2, 3]
aten::expand5
695Tensor<[7,1,1]>,
dims: [1, 2, 3]
aten::index.Tensor4
696Tensor<[1,1]>,
dims: [2, 3]
aten::index.Tensor4
697Tensor<[1,71,7,64]>,
dims: [0, 1, 2, 3]
aten::mul.Scalar4
698Tensor<[1,7,64]>,
dims: [0, 1, 2]
aten::mul.Tensor4
699Tensor<[1,16,112,112]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
700Tensor<[96]>,
dims: [0]
aten::add.Tensor4
701Tensor<[1,96,112,112]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
702Tensor<[96,1,1]>,
dims: [1, 2, 3]
aten::add.Tensor4
703Tensor<[1,96,56,56]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
704Tensor<[144]>,
dims: [0]
aten::add.Tensor4
705Tensor<[1,144,56,56]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
706Tensor<[144,1,1]>,
dims: [1, 2, 3]
aten::add.Tensor4
707Tensor<[1,144,28,28]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
708Tensor<[1,32,28,28]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
709Tensor<[1,192,28,28]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
710Tensor<[1,192,14,14]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
711Tensor<[1,64,14,14]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
712Tensor<[384]>,
dims: [0]
aten::add.Tensor4
713Tensor<[1,384,14,14]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
714Tensor<[384,1,1]>,
dims: [1, 2, 3]
aten::add.Tensor4
715Tensor<[1,96,14,14]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
716Tensor<[576]>,
dims: [0]
aten::add.Tensor4
717Tensor<[1,576,14,14]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
718Tensor<[576,1,1]>,
dims: [1, 2, 3]
aten::add.Tensor4
719Tensor<[1,576,7,7]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
720Tensor<[960]>,
dims: [0]
aten::add.Tensor4
721Tensor<[1,960,7,7]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
722Tensor<[960,1,1]>,
dims: [1, 2, 3]
aten::add.Tensor4
723Tensor<[1,320,7,7]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
724Tensor<[1,1280,7,7]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
725Tensor<[1,12,12,12]>,
dims: [0, 1, 2, 3]
aten::_safe_softmax4
726Tensor<[1,12,12,1]>,
dims: [0, 1, 2, 3]
aten::_safe_softmax4
727Tensor<[1,12,1]>,
dims: [0, 1, 2]
aten::add.Tensor4
728Tensor<[1,12,128]>,
dims: [0, 1, 2]
aten::add.Tensor4
729Tensor<[12,768]>,
dims: [0, 1]
aten::add.Tensor4
730Tensor<[1,1,12,12]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
731Tensor<[1,12,768]>,
dims: [0, 1, 2]
aten::add.Tensor4
732Tensor<[12,3072]>,
dims: [0, 1]
aten::add.Tensor4
733Tensor<[1,12,3072]>,
dims: [0, 1, 2]
aten::add.Tensor4
734Tensor<[12,2]>,
dims: [0, 1]
aten::add.Tensor4
735Tensor<[12,64,12]>,
dims: [0, 1, 2]
aten::bmm4
736Tensor<[12,12,64]>,
dims: [0, 1, 2]
aten::bmm4
737Tensor<[1,1,1,12]>,
dims: [0, 1, 2, 3]
aten::expand4
738Tensor<[1,12,12,64]>,
dims: [0, 1, 2, 3]
aten::mul.Scalar4
739Tensor<[1,12,64,12]>,
dims: [0, 1, 2, 3]
aten::mul.Scalar4
740Tensor<[1,12,9,9]>,
dims: [0, 1, 2, 3]
aten::_safe_softmax4
741Tensor<[1,12,9,1]>,
dims: [0, 1, 2, 3]
aten::_safe_softmax4
742Tensor<[1,9,1]>,
dims: [0, 1, 2]
aten::add.Tensor4
743Tensor<[1,9,128]>,
dims: [0, 1, 2]
aten::add.Tensor4
744Tensor<[9,768]>,
dims: [0, 1]
aten::add.Tensor4
745Tensor<[1,1,9,9]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
746Tensor<[1,9,768]>,
dims: [0, 1, 2]
aten::add.Tensor4
747Tensor<[9,3072]>,
dims: [0, 1]
aten::add.Tensor4
748Tensor<[1,9,3072]>,
dims: [0, 1, 2]
aten::add.Tensor4
749Tensor<[9,128]>,
dims: [0, 1]
aten::add.Tensor4
750Tensor<[9,30000]>,
dims: [0, 1]
aten::add.Tensor4
751Tensor<[30000]>,
dims: [1]
aten::add.Tensor4
752Tensor<[12,64,9]>,
dims: [0, 1, 2]
aten::bmm4
753Tensor<[12,9,64]>,
dims: [0, 1, 2]
aten::bmm4
754Tensor<[1,1,1,9]>,
dims: [0, 1, 2, 3]
aten::expand4
755Tensor<[1,12,9,64]>,
dims: [0, 1, 2, 3]
aten::mul.Scalar4
756Tensor<[1,12,64,9]>,
dims: [0, 1, 2, 3]
aten::mul.Scalar4
757Tensor<[30000]>,
dims: [0]
aten::mul.Tensor4
758Tensor<[1,16,9,9]>,
dims: [0, 1, 2, 3]
aten::_safe_softmax4
759Tensor<[1,16,9,1]>,
dims: [0, 1, 2, 3]
aten::_safe_softmax4
760Tensor<[9,2048]>,
dims: [0, 1]
aten::add.Tensor4
761Tensor<[1,9,2048]>,
dims: [0, 1, 2]
aten::add.Tensor4
762Tensor<[2048]>,
dims: [2]
aten::add.Tensor4
763Tensor<[9,8192]>,
dims: [0, 1]
aten::add.Tensor4
764Tensor<[8192]>,
dims: [1]
aten::add.Tensor4
765Tensor<[1,9,8192]>,
dims: [0, 1, 2]
aten::add.Tensor4
766Tensor<[16,128,9]>,
dims: [0, 1, 2]
aten::bmm4
767Tensor<[16,9,128]>,
dims: [0, 1, 2]
aten::bmm4
768Tensor<[1,16,9,128]>,
dims: [0, 1, 2, 3]
aten::mul.Scalar4
769Tensor<[1,16,128,9]>,
dims: [0, 1, 2, 3]
aten::mul.Scalar4
770Tensor<[8192]>,
dims: [0]
aten::mul.Tensor4
771Tensor<[9,1024]>,
dims: [0, 1]
aten::add.Tensor4
772Tensor<[1,9,1024]>,
dims: [0, 1, 2]
aten::add.Tensor4
773Tensor<[9,4096]>,
dims: [0, 1]
aten::add.Tensor4
774Tensor<[1,9,4096]>,
dims: [0, 1, 2]
aten::add.Tensor4
775Tensor<[16,64,9]>,
dims: [0, 1, 2]
aten::bmm4
776Tensor<[16,9,64]>,
dims: [0, 1, 2]
aten::bmm4
777Tensor<[1,16,9,64]>,
dims: [0, 1, 2, 3]
aten::mul.Scalar4
778Tensor<[1,16,64,9]>,
dims: [0, 1, 2, 3]
aten::mul.Scalar4
779Tensor<[1,64,9,9]>,
dims: [0, 1, 2, 3]
aten::_safe_softmax4
780Tensor<[1,64,9,1]>,
dims: [0, 1, 2, 3]
aten::_safe_softmax4
781Tensor<[9,16384]>,
dims: [0, 1]
aten::add.Tensor4
782Tensor<[16384]>,
dims: [1]
aten::add.Tensor4
783Tensor<[1,9,16384]>,
dims: [0, 1, 2]
aten::add.Tensor4
784Tensor<[64,64,9]>,
dims: [0, 1, 2]
aten::bmm4
785Tensor<[64,9,64]>,
dims: [0, 1, 2]
aten::bmm4
786Tensor<[1,64,9,64]>,
dims: [0, 1, 2, 3]
aten::mul.Scalar4
787Tensor<[1,64,64,9]>,
dims: [0, 1, 2, 3]
aten::mul.Scalar4
788Tensor<[16384]>,
dims: [0]
aten::mul.Tensor4
789Tensor<[1,2]>,
dims: [0, 1]
aten::add.Tensor4
790Tensor<[1,12,14,14]>,
dims: [0, 1, 2, 3]
aten::_safe_softmax4
791Tensor<[1,12,14,1]>,
dims: [0, 1, 2, 3]
aten::_safe_softmax4
792Tensor<[1,14,1]>,
dims: [0, 1, 2]
aten::add.Tensor4
793Tensor<[1,14,128]>,
dims: [0, 1, 2]
aten::add.Tensor4
794Tensor<[14,768]>,
dims: [0, 1]
aten::add.Tensor4
795Tensor<[1,1,14,14]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
796Tensor<[1,14,768]>,
dims: [0, 1, 2]
aten::add.Tensor4
797Tensor<[14,3072]>,
dims: [0, 1]
aten::add.Tensor4
798Tensor<[1,14,3072]>,
dims: [0, 1, 2]
aten::add.Tensor4
799Tensor<[14,2]>,
dims: [0, 1]
aten::add.Tensor4
800Tensor<[12,64,14]>,
dims: [0, 1, 2]
aten::bmm4
801Tensor<[12,14,64]>,
dims: [0, 1, 2]
aten::bmm4
802Tensor<[1,1,1,14]>,
dims: [0, 1, 2, 3]
aten::expand4
803Tensor<[1,12,14,64]>,
dims: [0, 1, 2, 3]
aten::mul.Scalar4
804Tensor<[1,12,64,14]>,
dims: [0, 1, 2, 3]
aten::mul.Scalar4
805Tensor<[1,12,50,50]>,
dims: [0, 1, 2, 3]
aten::_safe_softmax4
806Tensor<[1,12,50,1]>,
dims: [0, 1, 2, 3]
aten::_safe_softmax4
807Tensor<[2,8,7,7]>,
dims: [0, 1, 2, 3]
aten::_safe_softmax4
808Tensor<[2,8,7,1]>,
dims: [0, 1, 2, 3]
aten::_safe_softmax4
809Tensor<[1,50,1]>,
dims: [0, 1, 2]
aten::add.Tensor4
810Tensor<[1,50,768]>,
dims: [0, 1, 2]
aten::add.Tensor4
811Tensor<[50,768]>,
dims: [0, 1]
aten::add.Tensor4
812Tensor<[50,3072]>,
dims: [0, 1]
aten::add.Tensor4
813Tensor<[2,7,512]>,
dims: [0, 1, 2]
aten::add.Tensor4
814Tensor<[1,7,512]>,
dims: [0, 1, 2]
aten::add.Tensor4
815Tensor<[2,7,1]>,
dims: [0, 1, 2]
aten::add.Tensor4
816Tensor<[14,512]>,
dims: [0, 1]
aten::add.Tensor4
817Tensor<[2,1,7,7]>,
dims: [0, 1, 2, 3]
aten::add.Tensor4
818Tensor<[14,2048]>,
dims: [0, 1]
aten::add.Tensor4
819Tensor<[12,64,50]>,
dims: [0, 1, 2]
aten::bmm4
820Tensor<[12,50,64]>,
dims: [0, 1, 2]
aten::bmm4
821Tensor<[16,64,7]>,
dims: [0, 1, 2]
aten::bmm4
822Tensor<[16,7,64]>,
dims: [0, 1, 2]
aten::bmm4
823Tensor<[2,512]>,
dims: [0, 1]
aten::div.Tensor4
824Tensor<[2,1]>,
dims: [0, 1]
aten::div.Tensor4
825Tensor<[2,1,1,7]>,
dims: [0, 1, 2, 3]
aten::expand4
826Tensor<[1,12,50,64]>,
dims: [0, 1, 2, 3]
aten::mul.Scalar4
827Tensor<[1,12,64,50]>,
dims: [0, 1, 2, 3]
aten::mul.Scalar4
828Tensor<[2,8,7,64]>,
dims: [0, 1, 2, 3]
aten::mul.Scalar4
829Tensor<[2,8,64,7]>,
dims: [0, 1, 2, 3]
aten::mul.Scalar4
830Tensor<[1,50,3072]>,
dims: [0, 1, 2]
aten::mul.Tensor4
831Tensor<[2,7,2048]>,
dims: [0, 1, 2]
aten::mul.Tensor4
832Tensor<[1,16,197,197]>,
dims: [0, 1, 2, 3]
aten::_softmax4
833Tensor<[1,16,197,1]>,
dims: [0, 1, 2, 3]
aten::_softmax4
834Tensor<[1,197,1024]>,
dims: [0, 1, 2]
aten::add.Tensor4
835Tensor<[197,1024]>,
dims: [0, 1]
aten::add.Tensor4
836Tensor<[27]>,
dims: [0]
aten::add.Tensor4
837Tensor<[27,1]>,
dims: [0, 1]
aten::add.Tensor4
838Tensor<[196,196]>,
dims: [0, 1]
aten::add.Tensor4
839Tensor<[197,4096]>,
dims: [0, 1]
aten::add.Tensor4
840Tensor<[1,1024]>,
dims: [0, 1]
aten::add.Tensor4
841Tensor<[197]>,
dims: [0]
aten::arange4
842Tensor<[16,64,197]>,
dims: [0, 1, 2]
aten::bmm4
843Tensor<[16,197,64]>,
dims: [0, 1, 2]
aten::bmm4
844Tensor<[14,1]>,
dims: [0, 1]
aten::expand4
845Tensor<[1,14]>,
dims: [0, 1]
aten::expand4
846Tensor<[27,1]>,
dims: [2, 3]
aten::index.Tensor4
847Tensor<[27]>,
dims: [3]
aten::index.Tensor4
848Tensor<[1,16,27,27]>,
dims: [0, 1, 2, 3]
aten::mul.Tensor4
849Tensor<[2,196,1]>,
dims: [0, 1, 2]
aten::sub.Tensor4
850Tensor<[2,1,196]>,
dims: [0, 1, 2]
aten::sub.Tensor4
851Tensor<[1,197]>,
dims: [0, 1]
aten::where.self4
852Tensor<[196,197]>,
dims: [0, 1]
aten::where.self4
853Tensor<[197,1]>,
dims: [0, 1]
aten::where.self4
854Tensor<[197,197]>,
dims: [0, 1]
aten::where.self4
855Tensor<[12,1,1]>,
dims: [1, 2, 3]
aten::index.Tensor4
856Tensor<[1,12,27,27]>,
dims: [0, 1, 2, 3]
aten::mul.Tensor4
857Tensor<[1,64]>,
dims: [0, 1]
aten::add.Tensor4
858Tensor<[1,12]>,
dims: [0, 1]
aten::add.Tensor4
859Tensor<[12]>,
dims: [1]
aten::add.Tensor4
860Tensor<[1,784]>,
dims: [0, 1]
aten::add.Tensor4
861Tensor<[784]>,
dims: [1]
aten::add.Tensor4
862Tensor<[784]>,
dims: [0]
aten::mul.Tensor4