code: def __call__(self, axis_size, all_in_batched, *all_args):
      _, args = all_args
      consts_batched, in_batched = all_in_batched
      assert not any(
          tree_util.tree_leaves(consts_batched)
      ), consts_batched
      return call_rule(self.rule, axis_size, in_batched, args)
  
imports: from __future__ import annotations
  import functools
  import operator
  from typing import Callable
  from jax import lax
  from jax._src import api
  from jax._src import core
  from jax._src import custom_api_util
  from jax._src import linear_util
  from jax._src import source_info_util
  from jax._src import traceback_util
  from jax._src import tree_util
  from jax._src import util
  from jax._src.api_util import flatten_fun_nokwargs
  from jax._src.interpreters import ad
  from jax._src.interpreters import batching
  from jax._src.interpreters.batching import not_mapped
  from jax._src.interpreters import mlir
  from jax._src.interpreters import partial_eval
  from jax._src.interpreters import xla
  from jax._src.tree_util import tree_flatten, tree_map, tree_structure, tree_unflatten, treedef_tuple
label_desc: Use image node_id 2 for calling the ClosedRule obj's underlying member method code with example usage: obj.__call__(axis_size, all_in_batched) and returns: call_rule