code: def compute(
      self,
      layer: nn.Module,
      layer_args: List[Any],
      out: Any,
      module_name: str,
  ):
      if self.count_unique and module_name in self.seen_modules:
          return
      self.count += self.compute_fn(layer, layer_args, out)
      logging.debug(f"module name: {module_name}, count {self.count}")
      self.seen_modules.add(module_name)
  
imports: import collections.abc
  import logging
  import operator
  from typing import Any, Callable, Dict, List, Optional, Tuple, Union
  import torch
  import torch.nn
  from classy_vision.generic.util import eval_model, get_batchsize_per_replica, get_model_dummy_input, is_leaf, is_on_gpu
  from torch.cuda import cudart
label_desc: Use image node_id 2 for calling the ComplexityComputer obj's underlying member method code with example usage: obj.compute(layer, layer_args, out, module_name) without return types