From 7c7ca4c943cfa9be0ec1c6fe3f8ad37d83fbea38 Mon Sep 17 00:00:00 2001 From: "Steven S. Lyubomirsky" Date: Tue, 26 Oct 2021 21:04:16 -0700 Subject: [PATCH] Add TOpPattern flag to accelerator_call op (#17) --- src/relay/op/tensor/unary.cc | 1 + 1 file changed, 1 insertion(+) diff --git a/src/relay/op/tensor/unary.cc b/src/relay/op/tensor/unary.cc index 9922ca3d22125..c90ae3acbe333 100644 --- 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("AcceleratorFunc", "unknown") .set_attr>("OutputShape", {}) .set_support_level(10) + .set_attr("TOpPattern", kOpaque) .add_type_rel("AcceleratorCall", AcceleratorCallRel); RELAY_REGISTER_UNARY_OP("isnan")