Provide an option to create and feed the graph with tfrecords