def test_serialize_feature(self, feature, ascii_protos): serialized_features_tensor = tf_utils._serialize_feature(feature) with tf.compat.v1.Session(): serialized_features = serialized_features_tensor.eval() feature_proto = tf.train.Feature() self.assertEqual(len(serialized_features), len(ascii_protos)) for ascii_proto, serialized_feature in zip(ascii_protos, serialized_features): feature_proto.ParseFromString(serialized_feature) self.assertProtoEquals(ascii_proto, feature_proto)
def _serialize_feature(feature): return tf_utils._serialize_feature(feature)