class_docstr: Base test class and used for CPU case.
code: def _test_output(self, module):
      ref_output = self._compute_output(module)
  
      flat_module = FlattenParamsWrapper(module)
      flat_output = self._compute_output(flat_module)
      assert objects_are_equal(ref_output, flat_output)
  
imports: from collections import OrderedDict
  import unittest
  import torch
  from fairscale.fair_dev.testing.testing import objects_are_equal
  from fairscale.nn import FlattenParamsWrapper
label_desc: Use image node_id 11 for calling the TestFlattenParams obj's underlying member method code with example usage: obj._test_output(module) without return types