code: def dequantize(x, dq_dtype, scale):
      return x.astype(dq_dtype) * jnp.broadcast_to(
          scale.astype(dq_dtype), x.shape
      )
  
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 3 for calling a global function with example usage: dequantize(x, dq_dtype, scale) and returns: unknown