10 using ::Halide::MemoryType;
12 using ::Halide::TailStrategy;
65 Var n(
sum.get_schedule().dims()[0].var);
129 RVar r81_x(
f1.update(0).get_schedule().dims()[0].var);
139 .split(
n,
n,
ni, 8, TailStrategy::GuardWithIf)
145 .split(
n,
n,
ni, 8, TailStrategy::GuardWithIf)
150 .split(
n,
n,
ni, 8, TailStrategy::GuardWithIf)
183 .split(
v11,
v11,
v11i, 8, TailStrategy::GuardWithIf)
188 .split(
v11,
v11,
v11i, 8, TailStrategy::GuardWithIf)
223 .split(
v13,
v13,
v13i, 8, TailStrategy::GuardWithIf)
229 .reorder_storage(
v13,
v12);
267 .split(
v21,
v21,
v21i, 8, TailStrategy::GuardWithIf)
272 .split(
v21,
v21,
v21i, 8, TailStrategy::GuardWithIf)
276 .store_in(MemoryType::Stack)
277 .split(c, c,
ci, 8, TailStrategy::GuardWithIf)
280 .reorder(
ci, c, w,
n);
313 .split(
v22,
v22,
v22i, 8, TailStrategy::GuardWithIf)
318 .split(
v22,
v22,
v22i, 8, TailStrategy::GuardWithIf)
322 .split(w, w,
wi, 2, TailStrategy::GuardWithIf)
323 .split(
n,
n,
ni, 128, TailStrategy::GuardWithIf)
324 .split(c, c,
ci, 8, TailStrategy::GuardWithIf)
325 .split(
ni,
ni,
nii, 4, TailStrategy::GuardWithIf)
334 .store_in(MemoryType::Stack)
335 .split(c, c,
ci, 8, TailStrategy::GuardWithIf)
340 .reorder(
ci, c, w,
n);
342 .split(c, c,
ci, 8, TailStrategy::GuardWithIf)
370 .split(
v14,
v14,
v14i, 8, TailStrategy::GuardWithIf)
375 .split(
v14,
v14,
v14i, 8, TailStrategy::GuardWithIf)
419 .store_in(MemoryType::Stack)
420 .split(c, c,
ci, 8, TailStrategy::GuardWithIf)
423 .reorder(
ci, c,
s,
n);
425 .split(c, c,
ci, 8, TailStrategy::GuardWithIf)
429 .store_in(MemoryType::Stack)
430 .split(c, c,
ci, 8, TailStrategy::GuardWithIf)
434 .reorder(
ci, c, w,
s,
n);
436 .split(c, c,
ci, 8, TailStrategy::GuardWithIf)
442 .split(c, c,
ci, 8, TailStrategy::GuardWithIf)
447 .split(w, w,
wi, 8, TailStrategy::GuardWithIf)
453 .reorder_storage(w, c);
455 .split(w, w,
wi, 8, TailStrategy::GuardWithIf)
461 .split(c, c,
ci, 14, TailStrategy::GuardWithIf)
462 .split(
n,
n,
ni, 32, TailStrategy::GuardWithIf)
465 .split(w, w,
wi, 8, TailStrategy::GuardWithIf)
471 .reorder_storage(w, c,
n);
473 .split(
n,
n,
ni, 8, TailStrategy::GuardWithIf)
480 .store_in(MemoryType::Stack)
481 .split(c, c,
ci, 8, TailStrategy::GuardWithIf)
484 .reorder(
ci, c, w,
n);
488 .store_in(MemoryType::Stack)
489 .split(w, w,
wi, 8, TailStrategy::GuardWithIf)
494 .reorder_storage(w,
n);
496 .split(
n,
n,
ni, 8, TailStrategy::GuardWithIf)
504 .split(
n,
n,
ni, 8, TailStrategy::GuardWithIf)
510 .split(
n,
n,
ni, 8, TailStrategy::GuardWithIf)
515 .split(
n,
n,
ni, 128, TailStrategy::GuardWithIf)
516 .split(w, w,
wi, 2, TailStrategy::GuardWithIf)
517 .split(
ni,
ni,
nii, 8, TailStrategy::GuardWithIf)
524 .split(
n,
n,
ni, 512, TailStrategy::GuardWithIf)
525 .split(c, c,
ci, 10, TailStrategy::GuardWithIf)
526 .split(w, w,
wi, 2, TailStrategy::GuardWithIf)
527 .split(
ni,
ni,
nii, 8, TailStrategy::GuardWithIf)
534 .reorder_storage(
n, c, w);
536 .split(
n,
n,
ni, 512, TailStrategy::GuardWithIf)
537 .split(c, c,
ci, 10, TailStrategy::GuardWithIf)
538 .split(w, w,
wi, 2, TailStrategy::GuardWithIf)
539 .split(
ni,
ni,
nii, 8, TailStrategy::GuardWithIf)
546 .split(c, c,
ci, 3, TailStrategy::GuardWithIf)
547 .split(
n,
n,
ni, 8, TailStrategy::GuardWithIf)
550 .reorder(
ni,
n,
ci, c, w)
553 .reorder_storage(
n, c, w);
555 .split(
n,
n,
ni, 512, TailStrategy::GuardWithIf)
556 .split(c, c,
ci, 6, TailStrategy::GuardWithIf)
557 .split(
ni,
ni,
nii, 8, TailStrategy::GuardWithIf)
564 .reorder_storage(
n, c, w);
566 .split(
n,
n,
ni, 512, TailStrategy::GuardWithIf)
567 .split(c, c,
ci, 6, TailStrategy::GuardWithIf)
568 .split(
ni,
ni,
nii, 8, TailStrategy::GuardWithIf)
575 .split(c, c,
ci, 11, TailStrategy::GuardWithIf)
576 .split(
n,
n,
ni, 8, TailStrategy::GuardWithIf)
583 .split(w, w,
wi, 8, TailStrategy::GuardWithIf)
589 .reorder_storage(w, c);
591 .split(w, w,
wi, 8, TailStrategy::GuardWithIf)
597 .split(w, w,
wi, 8, TailStrategy::GuardWithIf)
603 .reorder_storage(w, c);
605 .split(w, w,
wi, 8, TailStrategy::GuardWithIf)
611 .split(
s,
s,
si, 8, TailStrategy::GuardWithIf)
614 .reorder(
si,
s, c,
n)
615 .reorder_storage(
s, c,
n);