Commit fe668828 authored by Martin Wicke's avatar Martin Wicke Committed by TensorFlower Gardener
Browse files

Refactor compatibility upgrade tool to use pasta (an AST based refactoring... 

Refactor compatibility upgrade tool to use pasta (an AST based refactoring tool) instead of line based edits.

This allows for some simplification (e.g., we can now remove arguments with a dict entry) and for much more powerful transformations.

This passes all tests with no destructive test changes (though there are cosmetic test changes).

Also adds transformations for tf.to_dtype -> tf.cast(..., dtype=...).

PiperOrigin-RevId: 227934801
parent 75f2e9c2
Expand all 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