10 using ::Halide::MemoryType;
12 using ::Halide::TailStrategy;
63 Var n(
sum.get_schedule().dims()[0].var);
130 RVar r24_x(
f1.update(0).get_schedule().dims()[0].var);
144 .split(
n,
n,
ni, 8, TailStrategy::GuardWithIf)
150 .split(
n,
n,
ni, 8, TailStrategy::GuardWithIf)
155 .split(
n,
n,
ni, 8, TailStrategy::GuardWithIf)
188 .split(
v2,
v2,
v2i, 8, TailStrategy::GuardWithIf)
193 .split(
v2,
v2,
v2i, 8, TailStrategy::GuardWithIf)
240 .split(
v4,
v4,
v4i, 8, TailStrategy::GuardWithIf)
245 .reorder_storage(
v4,
v3);
280 .split(
v12,
v12,
v12i, 8, TailStrategy::GuardWithIf)
285 .split(
v12,
v12,
v12i, 8, TailStrategy::GuardWithIf)
289 .store_in(MemoryType::Stack)
290 .split(c, c,
ci, 8, TailStrategy::GuardWithIf)
293 .reorder(
ci, c, w,
n);
333 .store_in(MemoryType::Stack)
334 .split(
v13,
v13,
v13i, 8, TailStrategy::GuardWithIf)
339 .split(
v13,
v13,
v13i, 8, TailStrategy::GuardWithIf)
343 .split(
n,
n,
ni, 5, TailStrategy::GuardWithIf)
344 .split(c, c,
ci, 8, TailStrategy::GuardWithIf)
347 .reorder(
ci,
ni, c, w,
n)
350 .store_in(MemoryType::Stack)
351 .split(c, c,
ci, 8, TailStrategy::GuardWithIf)
354 .reorder(
ci, c, w,
n);
356 .split(c, c,
ci, 8, TailStrategy::GuardWithIf)
382 .split(
v5,
v5,
v5i, 8, TailStrategy::GuardWithIf)
387 .split(
v5,
v5,
v5i, 8, TailStrategy::GuardWithIf)
434 .store_in(MemoryType::Stack)
435 .split(c, c,
ci, 8, TailStrategy::GuardWithIf)
438 .reorder(
ci, c,
s,
n);
440 .split(c, c,
ci, 8, TailStrategy::GuardWithIf)
444 .split(c, c,
ci, 8, TailStrategy::GuardWithIf)
450 .split(c, c,
ci, 8, TailStrategy::GuardWithIf)
455 .split(w, w,
wi, 8, TailStrategy::GuardWithIf)
460 .reorder_storage(w, c);
463 .split(w, w,
wi, 8, TailStrategy::GuardWithIf)
468 .store_in(MemoryType::Stack)
469 .split(w, w,
wi, 8, TailStrategy::GuardWithIf)
473 .reorder(
wi, w, c,
n)
474 .reorder_storage(w, c,
n);
476 .split(c, c,
ci, 2, TailStrategy::GuardWithIf)
477 .split(
n,
n,
ni, 8, TailStrategy::GuardWithIf)
480 .reorder(
ni,
n,
ci, w, c)
482 .reorder_storage(
n, c, w);
484 .split(
n,
n,
ni, 8, TailStrategy::GuardWithIf)
490 .split(
n,
n,
ni, 8, TailStrategy::GuardWithIf)
491 .split(w, w,
wi, 2, TailStrategy::GuardWithIf)
494 .reorder(
ni, c,
wi,
n, w)
497 .reorder_storage(
n, c, w);
499 .split(
n,
n,
ni, 8, TailStrategy::GuardWithIf)
500 .split(w, w,
wi, 2, TailStrategy::GuardWithIf)
502 .reorder(
ni,
wi,
n, w)
506 .split(
n,
n,
ni, 8, TailStrategy::GuardWithIf)
507 .split(w, w,
wi, 2, TailStrategy::GuardWithIf)
509 .reorder(
ni,
wi,
n, w)
513 .split(
n,
n,
ni, 8, TailStrategy::GuardWithIf)
514 .split(w, w,
wi, 2, TailStrategy::GuardWithIf)
516 .reorder(
ni,
wi,
n, w)
520 .split(
n,
n,
ni, 8, TailStrategy::GuardWithIf)
521 .split(w, w,
wi, 2, TailStrategy::GuardWithIf)
523 .reorder(
ni,
wi,
n, w)
527 .split(
n,
n,
ni, 8, TailStrategy::GuardWithIf)
528 .split(w, w,
wi, 2, TailStrategy::GuardWithIf)
530 .reorder(
ni,
wi,
n, w)
534 .split(
n,
n,
ni, 8, TailStrategy::GuardWithIf)
535 .split(w, w,
wi, 2, TailStrategy::GuardWithIf)
537 .reorder(
ni,
wi,
n, w)
541 .split(
n,
n,
ni, 8, TailStrategy::GuardWithIf)
542 .split(w, w,
wi, 2, TailStrategy::GuardWithIf)
544 .reorder(
ni,
wi,
n, w)
548 .split(
n,
n,
ni, 8, TailStrategy::GuardWithIf)
549 .split(w, w,
wi, 2, TailStrategy::GuardWithIf)
551 .reorder(
ni,
wi,
n, w)
555 .split(
n,
n,
ni, 8, TailStrategy::GuardWithIf)
556 .split(w, w,
wi, 2, TailStrategy::GuardWithIf)
558 .reorder(
ni,
wi,
n, w)
562 .split(
n,
n,
ni, 8, TailStrategy::GuardWithIf)
563 .split(w, w,
wi, 2, TailStrategy::GuardWithIf)
565 .reorder(
ni,
wi,
n, w)
569 .split(
n,
n,
ni, 8, TailStrategy::GuardWithIf)
570 .split(w, w,
wi, 2, TailStrategy::GuardWithIf)
572 .reorder(
ni,
wi,
n, w)
576 .split(
n,
n,
ni, 8, TailStrategy::GuardWithIf)
577 .split(w, w,
wi, 2, TailStrategy::GuardWithIf)
579 .reorder(
ni,
wi,
n, w)
583 .split(
n,
n,
ni, 8, TailStrategy::GuardWithIf)
584 .split(w, w,
wi, 2, TailStrategy::GuardWithIf)
586 .reorder(
ni,
wi,
n, w)
590 .split(
n,
n,
ni, 8, TailStrategy::GuardWithIf)
591 .split(w, w,
wi, 2, TailStrategy::GuardWithIf)
593 .reorder(
ni,
wi,
n, w)
597 .split(
n,
n,
ni, 8, TailStrategy::GuardWithIf)
598 .split(w, w,
wi, 2, TailStrategy::GuardWithIf)
600 .reorder(
ni,
wi,
n, w)
604 .split(
n,
n,
ni, 8, TailStrategy::GuardWithIf)
605 .split(w, w,
wi, 2, TailStrategy::GuardWithIf)
607 .reorder(
ni,
wi,
n, w)
611 .split(
n,
n,
ni, 8, TailStrategy::GuardWithIf)
612 .split(w, w,
wi, 2, TailStrategy::GuardWithIf)
614 .reorder(
ni,
wi,
n, w)
618 .split(
n,
n,
ni, 8, TailStrategy::GuardWithIf)
619 .split(w, w,
wi, 2, TailStrategy::GuardWithIf)
621 .reorder(
ni,
wi,
n, w)
625 .split(
n,
n,
ni, 8, TailStrategy::GuardWithIf)
626 .split(w, w,
wi, 2, TailStrategy::GuardWithIf)
628 .reorder(
ni,
wi,
n, w)
632 .split(
n,
n,
ni, 8, TailStrategy::GuardWithIf)
633 .split(w, w,
wi, 2, TailStrategy::GuardWithIf)
635 .reorder(
ni,
wi,
n, w)
639 .split(
n,
n,
ni, 8, TailStrategy::GuardWithIf)
640 .split(w, w,
wi, 2, TailStrategy::GuardWithIf)
642 .reorder(
ni,
wi,
n, w)
646 .split(
n,
n,
ni, 8, TailStrategy::GuardWithIf)
647 .split(w, w,
wi, 2, TailStrategy::GuardWithIf)
649 .reorder(
ni,
wi,
n, w)
653 .split(
n,
n,
ni, 8, TailStrategy::GuardWithIf)
654 .split(w, w,
wi, 2, TailStrategy::GuardWithIf)
656 .reorder(
ni,
wi,
n, w)
660 .split(
n,
n,
ni, 8, TailStrategy::GuardWithIf)
661 .split(w, w,
wi, 2, TailStrategy::GuardWithIf)
663 .reorder(
ni,
wi,
n, w)
667 .split(
n,
n,
ni, 8, TailStrategy::GuardWithIf)
668 .split(w, w,
wi, 2, TailStrategy::GuardWithIf)
670 .reorder(
ni,
wi,
n, w)
674 .split(
n,
n,
ni, 8, TailStrategy::GuardWithIf)
675 .split(w, w,
wi, 2, TailStrategy::GuardWithIf)
677 .reorder(
ni,
wi,
n, w)
681 .split(
n,
n,
ni, 8, TailStrategy::GuardWithIf)
682 .split(w, w,
wi, 2, TailStrategy::GuardWithIf)
684 .reorder(
ni,
wi,
n, w)
688 .split(
n,
n,
ni, 8, TailStrategy::GuardWithIf)
689 .split(w, w,
wi, 2, TailStrategy::GuardWithIf)
691 .reorder(
ni,
wi,
n, w)
695 .split(
n,
n,
ni, 8, TailStrategy::GuardWithIf)
696 .split(w, w,
wi, 2, TailStrategy::GuardWithIf)
698 .reorder(
ni,
wi,
n, w)
702 .split(
n,
n,
ni, 8, TailStrategy::GuardWithIf)
703 .split(w, w,
wi, 2, TailStrategy::GuardWithIf)
705 .reorder(
ni,
wi,
n, w)
709 .split(
n,
n,
ni, 8, TailStrategy::GuardWithIf)
710 .split(w, w,
wi, 2, TailStrategy::GuardWithIf)
712 .reorder(
ni,
wi,
n, w)
716 .split(
n,
n,
ni, 8, TailStrategy::GuardWithIf)
717 .split(w, w,
wi, 2, TailStrategy::GuardWithIf)
719 .reorder(
ni,
wi,
n, w)
723 .split(
n,
n,
ni, 8, TailStrategy::GuardWithIf)
724 .split(w, w,
wi, 2, TailStrategy::GuardWithIf)
727 .reorder(
ni,
wi,
n, w)
731 .split(
n,
n,
ni, 8, TailStrategy::GuardWithIf)
738 .split(
n,
n,
ni, 8, TailStrategy::GuardWithIf)
744 .split(
n,
n,
ni, 8, TailStrategy::GuardWithIf)
749 .split(c, c,
ci, 8, TailStrategy::GuardWithIf)
750 .split(w, w,
wi, 4, TailStrategy::GuardWithIf)
751 .split(
wi,
wi,
wii, 2, TailStrategy::GuardWithIf)
752 .split(
n,
n,
ni, 8, TailStrategy::GuardWithIf)
758 .reorder_storage(
n, c, w);
760 .split(c, c,
ci, 8, TailStrategy::GuardWithIf)
761 .split(w, w,
wi, 2, TailStrategy::GuardWithIf)
762 .split(
n,
n,
ni, 8, TailStrategy::GuardWithIf)
768 .split(c, c,
ci, 3, TailStrategy::GuardWithIf)
769 .split(w, w,
wi, 7, TailStrategy::GuardWithIf)
770 .split(
n,
n,
ni, 8, TailStrategy::GuardWithIf)
776 .reorder_storage(
n, c, w);
778 .split(
n,
n,
ni, 40, TailStrategy::GuardWithIf)
779 .split(c, c,
ci, 12, TailStrategy::GuardWithIf)
780 .split(w, w,
wi, 2, TailStrategy::GuardWithIf)
781 .split(
ni,
ni,
nii, 8, TailStrategy::GuardWithIf)
788 .reorder_storage(
n, c, w);
790 .split(
n,
n,
ni, 40, TailStrategy::GuardWithIf)
791 .split(c, c,
ci, 12, TailStrategy::GuardWithIf)
792 .split(w, w,
wi, 2, TailStrategy::GuardWithIf)
793 .split(
ni,
ni,
nii, 8, TailStrategy::GuardWithIf)
800 .split(c, c,
ci, 5, TailStrategy::GuardWithIf)
801 .split(
s,
s,
si, 7, TailStrategy::GuardWithIf)
802 .split(
n,
n,
ni, 8, TailStrategy::GuardWithIf)
809 .split(c, c,
ci, 8, TailStrategy::GuardWithIf)
814 .split(c, c,
ci, 8, TailStrategy::GuardWithIf)
818 .split(c, c,
ci, 8, TailStrategy::GuardWithIf)
824 .split(c, c,
ci, 8, TailStrategy::GuardWithIf)
829 .store_in(MemoryType::Stack)
830 .split(c, c,
ci, 8, TailStrategy::GuardWithIf)
833 .reorder(
ci, c, w,
s,
n);
835 .split(
s,
s,
si, 10, TailStrategy::GuardWithIf)
836 .split(
n,
n,
ni, 2, TailStrategy::GuardWithIf)
837 .split(c, c,
ci, 8, TailStrategy::GuardWithIf)