code: def quantize_dequantize(x, q_dtype, scale, compute_dtype):
      qx = quantize(x, q_dtype, scale, compute_dtype)
      return dequantize(qx, x.dtype, scale)
  
imports: import warnings
  from functools import partial
  from jax import custom_jvp, custom_vjp, lax, random
  from jax import numpy
  from flax.linen import initializers, module
label_desc: Use image node_id 4 for calling a global function with example usage: quantize_dequantize(x, q_dtype, scale, compute_dtype) and returns: dequantize