Missed TensorFlow World? Check out the recap. Learn more

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.