code: def ls_bert(model, inputs, attn_mask):
      torch.cuda.synchronize()
      start_time = time.perf_counter()
      ls_output = model.infer(inputs, attn_mask)
      torch.cuda.synchronize()
      end_time = time.perf_counter()
      return ls_output, end_time - start_time
  
file_path: lightseq/examples/inference/python/test/ls_bert.py
imports: import time
  import argparse
  import torch
  import lightseq.inference
  from transformers import BertTokenizer, BertForSequenceClassification
label_desc: Use image node_id 1 for calling a global function with example usage: ls_bert(model, inputs, attn_mask) and returns: ls_output,  unknown