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
Loading
Please sign in to comment