code: def in_qdq_fwd(compute_dtype, inp, scale, amax_history):
      qin, new_scale, new_history = qdq_and_return(
          inp, jnp.float8_e4m3fn, scale, amax_history, compute_dtype
      )
      return qin, (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 9 for calling a global function with example usage: in_qdq_fwd(compute_dtype, inp, scale, amax_history) and returns: qin,  new_scale,  new_history