Set op pattern to injective

apache · Apr 23, 2020 · c3297bc · c3297bc
1 parent d41d482
commit c3297bc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/tvm/relay/op/nn/_nn.py b/python/tvm/relay/op/nn/_nn.py
@@ -464,7 +464,7 @@ def compute_dilate(attrs, inputs, out_dtype):
     return [topi.nn.dilate(inputs[0], attrs.strides)]
 
 reg.register_broadcast_schedule("nn.dilate")
-reg.register_pattern("nn.dilate", OpPattern.OPAQUE)
+reg.register_pattern("nn.dilate", OpPattern.INJECTIVE)
 
 
 # cross_entropy_with_logits