tf.dtypes.as_dtype