code: def conv(
      lhs: Array,
      rhs: Array,
      window_strides: Sequence[int],
      padding: str,
      precision: lax.PrecisionLike = None,
      preferred_element_type: DTypeLike | None = None,
  ) -> Array:
      """Convenience wrapper around `conv_general_dilated`.
  
      Args:
        lhs: a rank `n+2` dimensional input array.
        rhs: a rank `n+2` dimensional array of kernel weights.
        window_strides: a sequence of `n` integers, representing the inter-window
          strides.
        padding: either the string `'SAME'`, the string `'VALID'`.
        precision: Optional. Either ``None``, which means the default precision for
          the backend, a :class:`~jax.lax.Precision` enum value (``Precision.DEFAULT``,
          ``Precision.HIGH`` or ``Precision.HIGHEST``) or a tuple of two
          :class:`~jax.lax.Precision` enums indicating precision of ``lhs``` and ``rhs``.
        preferred_element_type: Optional. Either ``None``, which means the default
          accumulation type for the input types, or a datatype, indicating to
          accumulate results to and return a result with that datatype.
  
      Returns:
        An array containing the convolution result.
      """
      return conv_general_dilated(
          lhs,
          rhs,
          window_strides,
          padding,
          precision=precision,
          preferred_element_type=preferred_element_type,
      )
  
imports: from __future__ import annotations
  from collections.abc import Sequence
  from functools import partial
  import operator
  from typing import NamedTuple, Union
  import numpy
  from jax._src import core
  from jax._src import dtypes
  from jax._src import util
  from jax._src.interpreters import ad
  from jax._src.interpreters import batching
  from jax._src.interpreters import mlir
  from jax._src.lax import lax
  from jax._src.lib.mlir.dialects import hlo
  from jax._src.typing import Array, DTypeLike
label_desc: Use image node_id 2 for calling a global function with example usage: conv(lhs, rhs, window_strides, padding, precision, preferred_element_type) and returns: conv_general_dilated