tf.dtypes.as_dtype

Converts the given type_value to a tf.DType.

tf.dtypes.as_dtype(
    type_value
)

Inputs can be existing tf.DType objects, a DataType enum, a string type name, or a numpy.dtype.

Examples:

tf.as_dtype(2)  # Enum value for float64.
tf.float64

tf.as_dtype('float')
tf.float32

tf.as_dtype(np.int32)
tf.int32

Note: DType values are interned (i.e. a single instance of each dtype is stored in a map). When passed a new DType object, as_dtype always returns the interned value.