class_docstr: Base test class and used for CPU case.
code: def _compute_output(self, module):
      device = next(module.parameters()).device
      dtype = next(module.parameters()).dtype
      input = module.get_input(device, dtype)
      return module(*input)
  
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 8 for calling the TestFlattenParams obj's underlying member method code with example usage: obj._compute_output(module) and returns: module