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, }