ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more


Updates dtypes of all specs in the input spec.

specs Nested list/tuple/dict of TensorSpecs/ArraySpecs, describing the shape of tensors.
dtype dtype to update the specs to.

Nested TensorSpecs with the udpated dtype.