class_docstr: Contains the information about a single row in the summary table.
  
  Attributes:
    path: A tuple of strings that represents the path to the module.
    module_type: type of the Module.
    method: method of the module called.
    inputs: inputs to the module.
    outputs: Output of the Module as reported by `capture_intermediates`.
    module_variables: Dictionary of variables in the module (no submodules
      included).
    counted_variables: Dictionary of variables that should be counted for this
      row, if no summarization is done (e.g. `depth=None` in `module_summary`)
      then this field is the same as `module_variables`, however if a
      summarization is done then this dictionary potentially contains parameters
      from submodules depending on the depth of the Module in question.
    flops: FLOPs cost of calling the module method.
    vjp_flops: FLOPs cost of calling the VJP of the module method.
class_docstr_tok: ["Contains","the","information","about","a","single","row","in","the","summary","table",".","Attributes",":","path",":","A","tuple","of","strings","that","represents","the","path","to","the","module",".","module_type",":","type","of","the","Module",".","method",":","method","of","the","module","called",".","inputs",":","inputs","to","the","module",".","outputs",":","Output","of","the","Module","as","reported","by","`","capture_intermediates","`",".","module_variables",":","Dictionary","of","variables","in","the","module","(","no","submodules","included",")",".","counted_variables",":","Dictionary","of","variables","that","should","be","counted","for","this","row",",","if","no","summarization","is","done","(","e.g",".","`","depth=None","`","in","`","module_summary","`",")","then","this","field","is","the","same","as","`","module_variables","`",",","however","if","a","summarization","is","done","then","this","dictionary","potentially","contains","parameters","from","submodules","depending","on","the","depth","of","the","Module","in","question",".","flops",":","FLOPs","cost","of","calling","the","module","method",".","vjp_flops",":","FLOPs","cost","of","calling","the","VJP","of","the","module","method","."]
code: def __post_init__(self):
      self.inputs = self.inputs
      self.outputs = self.outputs
      self.module_variables = self.module_variables
      self.counted_variables = self.counted_variables
  
imports: import dataclasses
  import enum
  import io
  from abc import ABC, abstractmethod
  from types import MappingProxyType
  from typing import Any, Callable, Dict, Iterable, List, Mapping, Optional, Sequence, Set, Tuple, Type, Union
  import jax
  import jax.numpy
  import numpy
  import rich.console
  import rich.table
  import rich.text
  import yaml
  import flax.linen.module
  from flax.core import meta, unfreeze
  from flax.core.scope import CollectionFilter, DenyList, FrozenVariableDict, LazyRng, MutableVariableDict
label_desc: Use image node_id 1 for calling the Row obj's underlying member method code with example usage: obj.__post_init__() without return types