Missed TensorFlow Dev Summit? Check out the video playlist. Watch recordings

tfl.premade_lib.build_lattice_layer

View source on GitHub

Creates a tfl.layers.Lattice layer.

tfl.premade_lib.build_lattice_layer(
    lattice_input, feature_configs, model_config, layer_output_range,
    submodel_index, is_inside_ensemble, dtype
)

Args:

  • lattice_input: Input to the lattice layer.
  • feature_configs: A list of tfl.configs.FeatureConfig instances that specify configurations for each feature.
  • model_config: Model configuration object describing model architecture. Should be one of the model configs in tfl.configs.
  • layer_output_range: A tfl.premade_lib.LayerOutputRange enum.
  • submodel_index: Corresponding index into submodels.
  • is_inside_ensemble: If this layer is inside an ensemble.
  • dtype: dtype

Returns:

A tfl.layers.Lattice instance.