code: def set_engine(self, stream, shape_dict):
      self.load()
      self.activate()
      self.stream = stream
      self.allocate_buffers(shape_dict, device="cuda")
  
imports: from collections import OrderedDict
  from copy import copy
  import numpy
  import tensorrt
  import torch
  from polygraphy import cuda
  from polygraphy.backend.common import bytes_from_path
  from polygraphy.backend.trt import engine_from_bytes
  from polygraphy.backend.trt import util
label_desc: Use image node_id 3 for calling the Engine obj's underlying member method code with example usage: obj.set_engine(stream, shape_dict) without return types