parameters: (sym, arg_params, aux_params)
return_statement: return arg_params, aux_params
docstring: 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
docstring_summary: 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
docstring_tokens: 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
function: 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
function_tokens: def init_tensorrt_params ( sym , arg_params , aux_params ) : 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