tf.contrib.autograph.set_element_type

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

Defined in tensorflow/python/autograph/lang/directives.py.

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.

Args:

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