Need to continue to use `tf.compat.v1` anyways due to placeholders and so on, so this may be a non issue.
