code: def test_export_import(self):
      path = self.create_tempdir().full_path
  
      m1 = Model()
      self.evaluate([v.initializer for v in m1.variables])
      self.evaluate(m1.scale.assign(m1.scale + 1.0))
  
      tf.saved_model.save(m1, os.path.join(path, "saved_model1"))
      m2 = tf.saved_model.load(os.path.join(path, "saved_model1"))
      self.evaluate(m2.scale.initializer)
      b = scale.Scale([5.0, 9.0], validate_args=True)
      self.evaluate(m2.make_bij(b).forward(2.0))
      self.evaluate(m2.scale.assign(m2.scale + [1.0, 2.0]))
      self.evaluate(m2.make_bij(b).forward(2.0))
  
      self.evaluate(m2.scale.assign([1.0, 2.0, 3.0]))
      tf.saved_model.save(m2, os.path.join(path, "saved_model2"))
      m3 = tf.saved_model.load(os.path.join(path, "saved_model2"))
      self.evaluate(m3.scale.initializer)
      with self.assertRaisesOpError("compatible shape"):
          self.evaluate(m3.make_bij(b).forward([3.0]))
  
imports: import functools
  import os
  from absl import flags
  from absl.testing import parameterized
  import tensorflow.compat.v2
  from tensorflow_probability.python.bijectors import bijector
  from tensorflow_probability.python.bijectors import reshape
  from tensorflow_probability.python.bijectors import scale
  from tensorflow_probability.python.bijectors import shift
  from tensorflow_probability.python.bijectors import transform_diagonal
  from tensorflow_probability.python.distributions import independent
  from tensorflow_probability.python.distributions import normal
  from tensorflow_probability.python.distributions import transformed_distribution
  from tensorflow_probability.python.experimental.distributions import mvn_precision_factor_linop
  from tensorflow_probability.python.internal import auto_composite_tensor
  from tensorflow_probability.python.internal import tensor_util
  from tensorflow_probability.python.internal import test_util
  from absl.testing import absltest
label_desc: Use image node_id 13 for calling the AutoCompositeTensorTest obj's underlying member method code with example usage: obj.test_export_import() without return types