code: def _batch_size(
      batch_size_per_device: int,
      device_count: int,
  ) -> int:
      return batch_size_per_device * device_count
  
file_path: scenic/scenic/projects/lang4video/configs/base.py
imports: from collections.abc import Sequence
  from typing import Literal
  from typing import Union
  import ml_collections
label_desc: Use image node_id 6 for calling a global function with example usage: _batch_size(batch_size_per_device, device_count) and returns: unknown