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

GaussianDropout & GaussianNoise are active only during training time. This can be skipped during inference.
alexwong · Feb 28, 2020 · 7303b7d · 7303b7d
1 parent 194a863
commit 7303b7d
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,
 }