code: def input_example(sz):
      mel = torch.randn(1, 1, 80, sz).cuda().half()
      z = torch.randn(1, 8, sz * 256 // 8, 1).cuda().half()
      return (
          mel,
          z,
      )
  
file_path: NeMo/tests/collections/tts/test_waveglow.py
imports: import os
  import tempfile
  import pytest
  import torch
  from omegaconf import DictConfig
  from nemo.collections.tts.models import WaveGlowModel
  from nemo.core.classes import typecheck
label_desc: Use image node_id 1 for calling a global function with example usage: input_example(sz) and returns: mel,  z