tensor_compute_op.cc
12.5 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