code: def _is_over_batch_size(graph, graph_batch_size):
      graph_size = _get_graph_size(graph)
      return any([x > y for x, y in zip(graph_size, graph_batch_size)])
  
imports: import functools
  from typing import Any, Callable, Generator, Iterable, Iterator, List, Mapping, Optional, Sequence, Union
  import jax
  from jax import lax
  import jax.numpy
  import jax.tree_util
  from jraph._src import graph
  import numpy
label_desc: Use image node_id 32 for calling a global function with example usage: _is_over_batch_size(graph, graph_batch_size) and returns: any