diff --git a/keras_core/initializers/initializer.py b/keras_core/initializers/initializer.py index fcd46d56d..825a362aa 100644 --- a/keras_core/initializers/initializer.py +++ b/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. diff --git a/keras_core/layers/input_spec.py b/keras_core/layers/input_spec.py index 2b2701178..b895d043c 100644 --- a/keras_core/layers/input_spec.py +++ b/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. diff --git a/keras_core/layers/layer.py b/keras_core/layers/layer.py index 09cb19348..5683490af 100644 --- a/keras_core/layers/layer.py +++ b/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) diff --git a/keras_core/losses/loss.py b/keras_core/losses/loss.py index 79e5b2d4a..e50693609 100644 --- a/keras_core/losses/loss.py +++ b/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__) diff --git a/keras_core/metrics/metric.py b/keras_core/metrics/metric.py index c0c5d4c1f..4b87b6972 100644 --- a/keras_core/metrics/metric.py +++ b/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__) diff --git a/keras_core/models/sequential.py b/keras_core/models/sequential.py index ec649d976..018548242 100644 --- a/keras_core/models/sequential.py +++ b/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 diff --git a/keras_core/optimizers/optimizer.py b/keras_core/optimizers/optimizer.py index a886d8739..2de337b57 100644 --- a/keras_core/optimizers/optimizer.py +++ b/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__( diff --git a/keras_core/regularizers/regularizers.py b/keras_core/regularizers/regularizers.py index 82d1f61f4..32b9f2e0c 100644 --- a/keras_core/regularizers/regularizers.py +++ b/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.