code: def init_tensorrt_params(sym, arg_params, aux_params):
      """
      Set weights in attributes of TensorRT nodes
      :param sym: Symbol, the symbol graph should contains some TensorRT nodes
      :param arg_params: arg_params
      :param aux_params: aux_params
      :return arg_params, aux_params: remaining params that are not in TensorRT nodes
      """
      arg_params = arg_params.copy()
      aux_params = aux_params.copy()
      for s in sym.get_internals():
          new_params_names = ""
          tensorrt_params = {}
          if "subgraph_params_names" in s.list_attr():
              keys = s.list_attr()["subgraph_params_names"].split(";")
              for k in keys:
                  if k in arg_params:
                      new_params_names += k + ";"
                      tensorrt_params[
                          "subgraph_param_" + k
                      ] = arg_params[k]
                      arg_params.pop(k)
                  elif k in aux_params:
                      new_params_names += k + ";"
                      tensorrt_params[
                          "subgraph_param_" + k
                      ] = aux_params[k]
                      aux_params.pop(k)
              new_attrs = {}
              for k, v in tensorrt_params.items():
                  new_attrs[k] = str(v.handle.value)
              if len(new_attrs) > 0:
                  s._set_attr(**new_attrs)
                  s._set_attr(
                      subgraph_params_names=new_params_names[:-1]
                  )
      return arg_params, aux_params
  
imports: import os
label_desc: Use image node_id 3 for calling a global function with example usage: init_tensorrt_params(sym, arg_params, aux_params) and returns: arg_params,  aux_params