Commit 4f787752 authored by Nick Felt's avatar Nick Felt Committed by TensorFlower Gardener
Browse files

Fix tf.contrib.summary migrations in tf_upgrade_v2

This addresses a few things I missed in https://github.com/tensorflow/tensorflow/commit/754b8a18b2a9aca1b53ef734b567b170ea3de00d:

- "tensor" kwarg got renamed to "data"
- "step" argument, if not originally passed, should be set to a default value because in TF 2.0 it's a mandatory argument. For now this uses tf.compat.v1.train.get_or_create_global_step() to preserve the same behavior as TF 1.x.

The pasta upgrade is required because the previous version has a bug that the code I added hits.

PiperOrigin-RevId: 234224028
parent 579fb173
Loading
Loading
Loading
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment