From 7303b7d79aab303ac7714b349ef6fc34859e77f1 Mon Sep 17 00:00:00 2001
From: Samuel <siju.samuel@huawei.com>
Date: Tue, 25 Feb 2020 09:09:10 +0530
Subject: [PATCH] [FRONTEND][KERAS]GaussianDropout/Noise parsing support
 (#4928)

GaussianDropout & GaussianNoise are active only during training time. This can be skipped during inference.
---
 python/tvm/relay/frontend/keras.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/python/tvm/relay/frontend/keras.py b/python/tvm/relay/frontend/keras.py
index caf41768ada4..adb28c466454 100644
--- a/python/tvm/relay/frontend/keras.py
+++ b/python/tvm/relay/frontend/keras.py
@@ -762,6 +762,8 @@ def _default_skip(inexpr, keras_layer, _): # pylint: disable=unused-argument
     'Dropout'                  : _default_skip,
     'SpatialDropout2D'         : _default_skip,
     'SpatialDropout1D'         : _default_skip,
+    'GaussianDropout'          : _default_skip,
+    'GaussianNoise'            : _default_skip,
 }