tf.layers.BaseLayer module

tf.layers.BaseLayer module

class tf.layers.BaseLayer.BaseLayer(*args: Any, **kwargs: Any)

Bases: tensorflow.keras.layers.Layer

Base layer for the reference models.

Parameters
  • boundary_casting (bool) – If True, outputs the values in half precision and casts the input values up to full precision.

  • tf_summary (bool) – If True, saves the activations with summary_layer.

call()