code: public static void registerCustomLayer(String layerName, Class<? extends KerasLayer> configClass) {
          customLayers.put(layerName, configClass);
      }
docs: Register a custom layer
  
  @param layerName   name of custom layer class
  @param configClass class of custom layer