code: def _batch_size_per_device(
      batch_size_per_device_memory_gb: int,
      device_memory_gbs: int,
  ) -> int:
      return batch_size_per_device_memory_gb * device_memory_gbs
  
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 5 for calling a global function with example usage: _batch_size_per_device(batch_size_per_device_memory_gb, device_memory_gbs) and returns: unknown