code: def scale_gradient(
      t: tf.Tensor, scale: types.FloatLike
  ) -> Tuple[tf.Tensor, types.GradFn]:
      """Scales gradients for the backwards pass.
  
      Args:
        t: A Tensor.
        scale: The scale factor for the gradient on the backwards pass.
  
      Returns:
        A Tensor same as input, with scaled backward gradient.
      """
  
      def grad(dy: tf.Tensor) -> Tuple[tf.Tensor, None]:
          """Scaled gradient."""
          return scale * dy, None
  
      return t, grad
  
label_desc: Use image node_id 1 for calling a global function with example usage: scale_gradient(t, scale) and returns: t,  grad, unknown,  None