View source on GitHub |
Module containing classes for transform.
Classes
class LayerNode
: Represents a Node in a tree containing a layer.
class LayerPattern
: Defines a tree sub-graph pattern of Keras layers to match in a model.
class Transform
: Defines a transform to be applied to a keras model graph.