docstr: Tests that mechanism for passing input shape works with static shape.
docstr_tok: ["Tests","that","mechanism","for","passing","input","shape","works","with","static","shape","."]
code: def test_decode_needs_input_shape_static(self):
      """Tests that mechanism for passing input shape works with static shape."""
      x = tf.reshape(list(range(15)), [3, 5])
      encoder = simple_encoder.SimpleEncoder(
          core_encoder.EncoderComposer(
              test_utils.ReduceMeanEncodingStage()
          ).make(),
          tf.TensorSpec.from_tensor(x),
      )
  
      state = encoder.initial_state()
      iteration = _make_iteration_function(encoder)
      _, _, decoded_x, _ = self.evaluate(iteration(x, state))
      self.assertAllEqual([[7.0] * 5] * 3, decoded_x)
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  from absl.testing import parameterized
  import tensorflow
  from tensorflow.python.framework import test_util
  from tensorflow_model_optimization.python.core.internal.tensor_encoding.core import core_encoder
  from tensorflow_model_optimization.python.core.internal.tensor_encoding.core import simple_encoder
  from tensorflow_model_optimization.python.core.internal.tensor_encoding.testing import test_utils
label_desc: Use image node_id 5 for calling the SimpleEncoderTest obj's underlying member method code with example usage: obj.test_decode_needs_input_shape_static() without return types