code: def qdq_and_return(x, q_dtype, scale, amax_history, compute_dtype):
      qx = quantize_dequantize(x, q_dtype, scale, compute_dtype)
      new_scale, new_history = compute_scale_and_amax_history(
          x, q_dtype, scale, amax_history
      )
      return qx, new_scale, new_history
  
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 7 for calling a global function with example usage: qdq_and_return(x, q_dtype, scale, amax_history, compute_dtype) and returns: qx,  new_scale,  new_history