tfm.nlp.serving_modules.features_to_int32

Stay organized with collections Save and categorize content based on your preferences.

Converts tf.int64 features to tf.int32, keep other features the same.

tf.Example only supports tf.int64, but the TPU only supports tf.int32.

features Input tensor dictionary.

Features with tf.int64 converted to tf.int32.