From 2798542e584164039f930cac5fac4d9f619e1610 Mon Sep 17 00:00:00 2001
From: Cody Yu <comaniac0422@gmail.com>
Date: Thu, 18 Mar 2021 00:25:08 +0000
Subject: [PATCH] [Relay] Add TopPattern to nn.dropout

---
 src/relay/op/nn/nn.cc | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/relay/op/nn/nn.cc b/src/relay/op/nn/nn.cc
index 0ea71de367fa..b2404cc1954b 100644
--- a/src/relay/op/nn/nn.cc
+++ b/src/relay/op/nn/nn.cc
@@ -590,6 +590,7 @@ The whole array is rescaled by ``1/(1-p)`` to keep the expected sum of the input
     .set_num_inputs(1)
     .add_argument("data", "Tensor", "Input to which dropout will be applied.")
     .set_support_level(1)
+    .set_attr<TOpPattern>("TOpPattern", kOpaque)
     .set_attr<FInferCorrectLayout>("FInferCorrectLayout", ElemwiseArbitraryLayout)
     .add_type_rel("Dropout", DropoutRel)
     .set_attr<TOpIsStateful>("TOpIsStateful", true);