tf.contrib.autograph.set_element_type

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.