code: def quantize_compute(
      tensor_A: te.Tensor, scale: float, zero_point: int, dtype: str
  ):
      """Compute for quantize"""
      scale_recip = 1 / scale
  
      return te.compute(
          tensor_A.shape,
          lambda n, h, w, c: saturate(
              (
                  (tensor_A[n, h, w, c] * scale_recip).astype("int32")
                  + zero_point
              ),
              dtype,
          ).astype(dtype),
          name="quantize",
      )
  
imports: from tvm import te
  from tvm import tir
  from ..utils import get_layout_transform_fn, saturate
label_desc: Use image node_id 1 for calling a global function with example usage: quantize_compute(tensor_A, scale, zero_point, dtype) and returns: te