Add export decorators for base classes
This commit is contained in:
parent
701d4cf7bc
commit
377fcdb91e
@ -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.
|
||||
|
||||
|
@ -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.
|
||||
|
||||
|
@ -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)
|
||||
|
@ -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__)
|
||||
|
@ -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__)
|
||||
|
@ -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
|
||||
|
@ -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__(
|
||||
|
@ -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.
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user