code: def format_expression(exp: Expression, depth: int) -> str:
      if exp.submodule:
          return format_module(exp.submodule, depth)
      s = "  " * depth + exp.primitive
      if exp.flops is not None:
          s += " " + _decimal_prefix(exp.flops, "flops")
      s += " " + exp.details
      return s + "\n"
  
imports: from collections.abc import Mapping, Sequence
  import dataclasses
  import itertools
  import logging
  import os
  import sys
  from typing import Any, Callable, NamedTuple, Optional
  from haiku._src import summarise
  import jax
  import jax.core
label_desc: Use image node_id 12 for calling a global function with example usage: format_expression(exp, depth) and returns: unknown, format_module