code: def test_with_multiple_input_shapes_pad_1_0(
      self, input_dims, expected_output_dims
  ):
      stage = kashin.KashinHadamardEncodingStage(
          pad_extra_level_threshold=1.0
      )
      test_data = self.run_one_to_many_encode_decode(
          stage, lambda: tf.random.normal(input_dims)
      )
      self.common_asserts_for_test_data(test_data)
      # Make sure output shape is as expected.
      self.assertEqual(
          expected_output_dims,
          test_data.encoded_x[
              kashin.KashinHadamardEncodingStage.ENCODED_VALUES_KEY
          ].shape,
      )
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import itertools
  from absl.testing import parameterized
  import numpy
  import tensorflow
  from tensorflow_model_optimization.python.core.internal.tensor_encoding.stages.research import kashin
  from tensorflow_model_optimization.python.core.internal.tensor_encoding.testing import test_utils
label_desc: Use image node_id 6 for calling the KashinHadamardEncodingStageTest obj's underlying member method code with example usage: obj.test_with_multiple_input_shapes_pad_1_0(input_dims, expected_output_dims) without return types