[FRONTEND][KERAS]GaussianDropout/Noise parsing support (#4928)

GaussianDropout & GaussianNoise are active only during training time. This can be skipped during inference.
apache · Feb 25, 2020 · 13cf1da · 13cf1da
1 parent 623dd20
commit 13cf1da
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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,
 }