tree 174221dacf28ba81688ad6197d12428fdff29cae
parent 3b6cf4f210a70217adc9913621512d0444da76d5
author Ken Franko <kfranko@google.com> 1575671483 -0800
committer TensorFlower Gardener <gardener@tensorflow.org> 1575672896 -0800

Group variable initialization when calling lift_to_graph.

When initializing variables defined inside a @tf.function which are lifted to the outer graph, group the variables together and call lift_to_graph once.  lift_to_graph supports passing in multiple tensors and the graph to lift to is the same for all of the variable initialization.  This improves setup time.

PiperOrigin-RevId: 284263511
Change-Id: I4cfcdb0394198df8f890a98295cc2fcb77b75413