code: def shape_dtype_to_ir_type(shape: Sequence[int], dtype) -> ir.Type:
      return ir.RankedTensorType.get(shape, dtype_to_ir_type(dtype))
  
file_path: jax/jaxlib/hlo_helpers.py
imports: from __future__ import annotations
  from collections.abc import Sequence
  from functools import partial
  from typing import Callable, Union
  import jaxlib.mlir.ir
  import jaxlib.mlir.dialects.stablehlo
  import numpy
label_desc: Use image node_id 2 for calling a global function with example usage: shape_dtype_to_ir_type(shape, dtype) and returns: ir