Add export decorators for base classes

keras_core/initializers/initializer.py

@@ -1,6 +1,8 @@
 import warnings
+from keras_core.api_export import keras_core_export
 
 
+@keras_core_export(["keras_core.Initializer", "keras_core.initializers.Initializer"])
 class Initializer:
     """Initializer base class: all Keras initializers inherit from this class.
 

keras_core/layers/input_spec.py

@@ -1,7 +1,9 @@
 from keras_core import backend
 from tensorflow import nest
+from keras_core.api_export import keras_core_export
 
 
+@keras_core_export(["keras_core.InputSpec", "keras_core.layers.InputSpec"])
 class InputSpec:
     """Specifies the rank, dtype and shape of every input to a layer.
 

keras_core/layers/layer.py

@@ -21,6 +21,7 @@ from keras_core.utils.tracking import Tracker
 from keras_core.metrics.metric import Metric
 from keras_core import utils
 from keras_core.layers import input_spec
+from keras_core.api_export import keras_core_export
 from tensorflow import nest
 from tensorflow import keras as tf_keras
 import numpy as np
@@ -29,6 +30,7 @@ import threading
 import collections
 
 
+@keras_core_export(["keras_core.Layer", "keras_core.layers.Layer"])
 class Layer(Operation):
     def __init__(self, trainable=True, dtype=None, name=None):
         super().__init__(name=name)

keras_core/losses/loss.py

@@ -1,8 +1,10 @@
 from keras_core import operations as ops
 from keras_core import backend
 from keras_core.utils.naming import auto_name
+from keras_core.api_export import keras_core_export
 
 
+@keras_core_export(["keras_core.Loss", "keras_core.losses.Loss"])
 class Loss:
     def __init__(self, name=None, reduction="sum_over_batch_size"):
         self.name = name or auto_name(self.__class__.__name__)

keras_core/metrics/metric.py

@@ -1,8 +1,10 @@
 from keras_core import backend
 from keras_core.utils.tracking import Tracker
 from keras_core.utils.naming import auto_name
+from keras_core.api_export import keras_core_export
 
 
+@keras_core_export(["keras_core.Metric", "keras_core.metrics.Metric"])
 class Metric:
     def __init__(self, dtype=None, name=None):
         self.name = name or auto_name(self.__class__.__name__)

keras_core/models/sequential.py

@@ -1,5 +1,8 @@
+from keras_core.api_export import keras_core_export
+from keras_core.models.model import Model
 
 
+@keras_core_export(["keras_core.Sequential", "keras_core.models.Sequential"])
 class Sequential(Model):
     def __init__(self, layers, trainable=True, name=None):
         pass

keras_core/optimizers/optimizer.py

@@ -4,10 +4,12 @@ from keras_core.utils.tracking import Tracker
 from keras_core import initializers
 from keras_core.optimizers.schedules import learning_rate_schedule
 from keras_core.utils.naming import auto_name
+from keras_core.api_export import keras_core_export
 import re
 import warnings
 
 
+@keras_core_export(["keras_core.Optimizer", "keras_core.optimizers.Optimizer"])
 class Optimizer:
     # TODO: support jit_compile
     def __init__(

keras_core/regularizers/regularizers.py

@@ -3,7 +3,7 @@ from keras_core import operations as ops
 from keras_core.api_export import keras_core_export
 
 
-@keras_core_export("keras_core.regularizers.Regularizer")
+@keras_core_export(["keras_core.Regularizer", "keras_core.regularizers.Regularizer"])
 class Regularizer:
     """Regularizer base class.