Rename tf.colocate_with to tf.compat.v1.colocate_with in upgrade script.
tf.colocate_with is no longer supported as public API in v2. In the upgrade script convert it to tf.compat.v1.colocate_with. PiperOrigin-RevId: 220166863
Loading
Please sign in to comment