deploy model with custom layer

[Tixierae]

Hi, and thank you for the great package!

when using export_savedmodel, which is the first step listed on the CloudML Deploying Models step-by-step instructions, I receive an error ValueError: Unknown layer: AttentionWithContext. AttentionWithContext is a custom layer that my model is using.

Since many users use custom layers, it would be very helpful to address this issue. Maybe using an approach similar to, e.g., Keras' custom_objects? See https://keras.io/getting-started/faq/#handling-custom-layers-or-other-custom-objects-in-saved-models

If a solution already exists, any help/advice would be highly appreciated.