code: def ls_generate(model, inputs_id, attn_mask):
      print("=========lightseq=========")
      print("lightseq generating...")
      ls_output, ls_time = ls_bert(model, inputs_id, attn_mask)
      print(f"lightseq time: {ls_time}s")
      print("lightseq results (class predictions):")
      print(ls_output.argmax(axis=1).detach().cpu().numpy())
  
file_path: lightseq/examples/inference/python/test/ls_bert_bench.py
imports: import time
  import argparse
  import torch
  import lightseq.inference
  from transformers import BertTokenizer, BertForSequenceClassification, BertModel
label_desc: Use image node_id 3 for calling a global function with example usage: ls_generate(model, inputs_id, attn_mask) without return types