[SCHEDULE] Add factor_axis to rfactor (#895)
Showing
- include/tvm/schedule.h 3 additions, 1 deletioninclude/tvm/schedule.h
- python/tvm/schedule.py 4 additions, 2 deletionspython/tvm/schedule.py
- src/api/api_lang.cc 1 addition, 1 deletionsrc/api/api_lang.cc
- src/schedule/schedule_dataflow_rewrite.cc 23 additions, 7 deletionssrc/schedule/schedule_dataflow_rewrite.cc
- tests/python/integration/test_reduce.py 31 additions, 0 deletionstests/python/integration/test_reduce.py
- tests/python/unittest/test_lang_schedule.py 10 additions, 0 deletionstests/python/unittest/test_lang_schedule.py
Loading
Please register or sign in to comment