Skip to content

T
tic
Commit 3b91ae28 by Siju Committed by Tianqi Chen
Options

[RELAY]missing schedules updated (#2196)

parent 8fe38eef
Showing with 2 additions and 1 deletions
python/tvm/relay/op/_tensor.py
......@@ -20,6 +20,7 @@ register_schedule("round", schedule_broadcast)
register_schedule("abs", schedule_broadcast)
register_schedule("tanh", schedule_broadcast)
register_schedule("negative", schedule_broadcast)
register_schedule("copy", schedule_broadcast)
register_schedule("add", schedule_broadcast)
register_schedule("subtract", schedule_broadcast)
......@@ -33,7 +34,7 @@ register_schedule("less", schedule_broadcast)
register_schedule("less_equal", schedule_broadcast)
register_schedule("greater", schedule_broadcast)
register_schedule("greater_equal", schedule_broadcast)
register_schedule("maximum_compute", schedule_injective)
register_schedule("maximum", schedule_injective)
register_schedule("minimum", schedule_injective)
register_schedule("right_shift", schedule_injective)
register_schedule("left_shift", schedule_injective)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment