compute_op.cc
19 KB
-
[TVM][BUGFIX] Fix missing reduction init predicates (#2495) · f0ae8e31* [TVM][BUGFIX] Fix reductions in split axes * A test case for the problem * Fix the fix: skip loops that are related to reduction AND are unrelated to axis
Sergei Grechanik committed