36 Func squashed_head1_filter_0_d_def__ = pipeline.
get_func(33);
37 Func squashed_head1_filter_broadcast_0_d_def__ = pipeline.
get_func(32);
56 Func normalized_schedule_features = pipeline.
get_func(12);
59 Func squashed_head1_filter_broadcast = pipeline.
get_func(6);
63 Var n(
sum.get_schedule().dims()[0].var);
118 RVar r1029_xi(
"r1029$xi");
132 RVar r34_x(
sum.update(0).get_schedule().dims()[0].var);
133 RVar r34_y(
sum.update(0).get_schedule().dims()[1].var);
152 .reorder(ni, r34_x, r34_y, n)
165 .
fuse(v221, v222, v221)
170 .reorder(v221i, v221)
175 .reorder(v221i, v221)
180 .reorder(v221i, v221)
185 .reorder(v221i, v221)
192 bias1_im_0_d_def__.
update(0)
195 .reorder(v2i, v2, r1226_x);
203 .
reorder(v218ii, v218i, v219i, v220i, v218, v219, v220)
204 .
fuse(v219, v220, v219)
205 .
fuse(v218, v219, v218)
212 .reorder(v218ii, v218i, v219i, v218, v219)
213 .fuse(v218, v219, v218)
220 .reorder(v218ii, v218i, v219i, v218, v219)
221 .fuse(v218, v219, v218)
228 .reorder(v218ii, v218i, v219i, v218, v219)
229 .fuse(v218, v219, v218)
236 .reorder(v218ii, v218i, v219i, v218, v219)
237 .fuse(v218, v219, v218)
245 .reorder_storage(v4, v3);
246 filter1_im_0_d_def__.
update(0)
247 .
reorder(r1010_x, r1010_y, r1010_z, v3)
249 filter1_im_0_d_def__.
update(1)
257 .
fuse(v216, v217, v216)
259 updated_head2_bias.
update(0)
262 .reorder(v216i, v216)
264 updated_head2_bias.
update(1)
267 .reorder(v216i, v216)
269 updated_head2_bias.
update(2)
272 .reorder(v216i, v216)
274 updated_head2_bias.
update(3)
277 .reorder(v216i, v216)
279 head2_bias_im_0_d_def__
284 head2_bias_im_0_d_def__.
update(0)
287 .reorder(v12i, v12, r1114_x, r1114_y);
288 head2_conv_0_d_def___1
300 .
reorder(v213i, v214i, v215i, v213, v214, v215)
301 .
fuse(v214, v215, v214)
302 .
fuse(v213, v214, v213)
304 updated_head2_filter.
update(0)
308 .reorder(v213i, v214i, v213, v214)
309 .fuse(v213, v214, v213)
311 updated_head2_filter.
update(1)
315 .reorder(v213i, v214i, v213, v214)
316 .fuse(v213, v214, v213)
318 updated_head2_filter.
update(2)
322 .reorder(v213i, v214i, v213, v214)
323 .fuse(v213, v214, v213)
325 updated_head2_filter.
update(3)
329 .reorder(v213i, v214i, v213, v214)
330 .fuse(v213, v214, v213)
332 head2_filter_im_0_d_def__
338 head2_filter_im_0_d_def__.
update(0)
341 .reorder(v13i, v13, v14, r1095_x, r1095_y);
355 head2_relu_0_d_def__.
update(0)
358 .reorder(ci, c, w, n, r986_x);
365 updated_head1_bias.
update(0)
368 .reorder(v211i, v211);
369 updated_head1_bias.
update(1)
372 .reorder(v211i, v211);
373 updated_head1_bias.
update(2)
376 .reorder(v211i, v211);
377 updated_head1_bias.
update(3)
380 .reorder(v211i, v211);
381 head1_bias_im_0_d_def__
386 head1_bias_im_0_d_def__.
update(0)
389 .reorder(v5i, v5, r1302_x);
397 .
reorder(v207i, v207, v208i, v209i, v210i, v208, v209, v210)
398 .
fuse(v209, v210, v209)
399 .
fuse(v208, v209, v208)
401 updated_head1_filter.
update(0)
406 .reorder(v207i, v207, v208i, v209i, v208, v209)
407 .fuse(v208, v209, v208)
409 updated_head1_filter.
update(1)
414 .reorder(v207i, v207, v208i, v209i, v208, v209)
415 .fuse(v208, v209, v208)
417 updated_head1_filter.
update(2)
422 .reorder(v207i, v207, v208i, v209i, v208, v209)
423 .fuse(v208, v209, v208)
425 updated_head1_filter.
update(3)
430 .reorder(v207i, v207, v208i, v209i, v208, v209)
431 .fuse(v208, v209, v208)
433 squashed_head1_filter_0_d_def__
439 squashed_head1_filter_0_d_def__.
update(0)
442 .reorder(ci, c, s, n, r1321_x);
449 head1_conv_1_d_def__.
update(0)
452 .reorder(ci, c, r1183_x, w)
454 conv1_stage1_1_d_def__
460 .reorder_storage(w, c);
461 conv1_stage1_1_d_def__.
update(0)
465 .reorder(wi, r1029_xi, r1029_x, w, c)
467 conv1_stage2_0_d_def___1
472 .
store_at(conv1_stage1_1_d_def__, r1029_x)
474 .reorder_storage(w, c, n);
475 conv1_stage2_1_d_def__
482 .reorder_storage(n, c, w);
497 .reorder_storage(n, c, w);
502 .reorder(ni, wi, n, w)
509 .reorder(ni, wi, n, w)
516 .reorder(ni, wi, n, w)
523 .reorder(ni, wi, n, w)
530 .reorder(ni, wi, n, w)
537 .reorder(ni, wi, n, w)
544 .reorder(ni, wi, n, w)
551 .reorder(ni, wi, n, w)
558 .reorder(ni, wi, n, w)
565 .reorder(ni, wi, n, w)
568 relu1_0_d_def__.
update(10)
572 .reorder(ni, wi, n, w)
575 relu1_0_d_def__.
update(11)
579 .reorder(ni, wi, n, w)
582 relu1_0_d_def__.
update(12)
586 .reorder(ni, wi, n, w)
589 relu1_0_d_def__.
update(13)
593 .reorder(ni, wi, n, w)
596 relu1_0_d_def__.
update(14)
600 .reorder(ni, wi, n, w)
603 relu1_0_d_def__.
update(15)
607 .reorder(ni, wi, n, w)
610 relu1_0_d_def__.
update(16)
614 .reorder(ni, wi, n, w)
617 relu1_0_d_def__.
update(17)
621 .reorder(ni, wi, n, w)
624 relu1_0_d_def__.
update(18)
628 .reorder(ni, wi, n, w)
631 relu1_0_d_def__.
update(19)
635 .reorder(ni, wi, n, w)
638 relu1_0_d_def__.
update(20)
642 .reorder(ni, wi, n, w)
645 relu1_0_d_def__.
update(21)
649 .reorder(ni, wi, n, w)
652 relu1_0_d_def__.
update(22)
656 .reorder(ni, wi, n, w)
659 relu1_0_d_def__.
update(23)
663 .reorder(ni, wi, n, w)
666 relu1_0_d_def__.
update(24)
670 .reorder(ni, wi, n, w)
673 relu1_0_d_def__.
update(25)
677 .reorder(ni, wi, n, w)
680 relu1_0_d_def__.
update(26)
684 .reorder(ni, wi, n, w)
687 relu1_0_d_def__.
update(27)
691 .reorder(ni, wi, n, w)
694 relu1_0_d_def__.
update(28)
698 .reorder(ni, wi, n, w)
701 relu1_0_d_def__.
update(29)
705 .reorder(ni, wi, n, w)
708 relu1_0_d_def__.
update(30)
712 .reorder(ni, wi, n, w)
715 relu1_0_d_def__.
update(31)
719 .reorder(ni, wi, n, w)
746 .reorder(ni, r24_x, n)
755 .
reorder(ni, n, ci, wii, wi, c, w)
758 .reorder_storage(n, c, w);
764 .reorder(ni, r19_x, n, ci, wi, c, w)
776 .reorder_storage(n, c, w);
784 .
reorder(nii, ni, ci, wi, n, c, w)
788 .reorder_storage(n, c, w);
795 .reorder(nii, r9_x, ni, ci, wi, n, c, w)
799 normalized_schedule_features
816 .reorder(ci, c, w, r14_x);
826 .reorder(ci, c, r4_x, r4_y, w)
828 squashed_head1_filter_broadcast
834 squashed_head1_filter