Commit ce5a8d8f authored by Allen Lavoie's avatar Allen Lavoie Committed by TensorFlower Gardener
Browse files

Allow users to interact with resource variables in imported 1.x SavedModels 

Modifies wrap_function to search for unlifted variables and lift them out as captures. Adds a test for training an imported model.

PiperOrigin-RevId: 233142944
parent 1798f307
Show whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment