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. |