Is it possible to specify the a tensor’s dtype in the typing hint of a function? We use input_signature to declare the signature of a tf.function,
input_signature
tf.function