tf.contrib.autograph.set_element_type

View source on GitHub

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

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

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