import tensorflow as tf

              row_splits=[0, 4, 4, 7, 8, 8]),
      }
      with tf.compat.v1.Session(graph=graph) as session:
        schema = schema_inference.infer_feature_schema(outputs, graph, session)
