Add TOpPattern flag to accelerator_call op (apache#17)

gussmith23 · Dec 29, 2021 · 7c7ca4c · 7c7ca4c
1 parent ce0ad7d
commit 7c7ca4c
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/relay/op/tensor/unary.cc b/src/relay/op/tensor/unary.cc
@@ -534,6 +534,7 @@ RELAY_REGISTER_OP("accelerator_call")
     .set_attr<String>("AcceleratorFunc", "unknown")
     .set_attr<Array<Integer>>("OutputShape", {})
     .set_support_level(10)
+    .set_attr<TOpPattern>("TOpPattern", kOpaque)
     .add_type_rel("AcceleratorCall", AcceleratorCallRel);
 
 RELAY_REGISTER_UNARY_OP("isnan")