TF 2.0 is out! Get hands-on practice at TF World, Oct 28-31. Use code TF20 for 20% off select passes. Register now

tf.contrib.autograph.set_element_type

View source on GitHub

Indicates that the entity is expected hold items of specified type/shape.

tf.contrib.autograph.set_element_type(
    entity,
    dtype,
    shape=UNSPECIFIED
)

The staged TensorFlow ops will reflect and assert this data type. Ignored otherwise.

Args:

  • entity: The entity to annotate.
  • dtype: TensorFlow dtype value to assert for entity.
  • shape: Optional shape to assert for entity.