class_docstr: Tests for `_tf_style_*` decorators.
  
  The class decorators, `tf_style_encoding_stage` and
  `tf_style_adaptive_encoding_stage` are tested by the example implementations
  of `EncodingStageInterface` and `AdaptiveEncodingStageInterface`, in
  `test_utils.py`.
class_docstr_tok: ["Tests","for","`","_tf_style_","*","`","decorators",".","The","class","decorators",",","`","tf_style_encoding_stage","`","and","`","tf_style_adaptive_encoding_stage","`","are","tested","by","the","example","implementations","of","`","EncodingStageInterface","`","and","`","AdaptiveEncodingStageInterface","`",",","in","`","test_utils.py","`","."]
docstr: Returns decorated test `decode` method.
docstr_tok: ["Returns","decorated","test","`","decode","`","method","."]
code: def _test_decode_fn(self):
      """Returns decorated test `decode` method."""
      reshape_fn = lambda x, p, shape: tf.reshape(
          x["val"] + p["param"], shape
      )
      return encoding_stage._tf_style_decode(
          lambda _, x, p, num_summands, shape, name: reshape_fn(
              x, p, shape
          )
      )
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  from absl.testing import parameterized
  import mock
  import numpy
  import tensorflow
  from tensorflow_model_optimization.python.core.internal.tensor_encoding.core import encoding_stage
  from tensorflow_model_optimization.python.core.internal.tensor_encoding.testing import test_utils
label_desc: Use image node_id 7 for calling the TFStyleEncodeDecodeTest obj's underlying member method code with example usage: obj._test_decode_fn() and returns: encoding_stage