code: def on_per_batch_transform_on_device(
      self, batch: Any, running_stage: RunningStage
  ) -> None:
      self.run_for_all_callbacks(
          batch,
          running_stage,
          method_name="on_per_batch_transform_on_device",
      )
  
imports: from contextlib import contextmanager
  from typing import Any, List, Sequence
  from pytorch_lightning.callbacks import Callback
  from torch import Tensor
  import flash
  from flash.core.data.utils import _STAGES_PREFIX
  from flash.core.utilities.stages import RunningStage
label_desc: Use image node_id 8 for calling the ControlFlow obj's underlying member method code with example usage: obj.on_per_batch_transform_on_device(batch, running_stage) without return types