Method is_trt_wrapper

tensorrt_llm/functional.py:575–584

Check if there is a trt.ITensor member inside, which is required for graph rewriter. In order to differentiate usages, it may be necessary to have an inheritance hierarchy.

(self)

            flayer and flayer.replace_input_with(self, new_tensor)

    def is_trt_wrapper(self):
        '''
        Check if there is a trt.ITensor member inside, which is required for
        graph rewriter. In order to differentiate usages, it may be necessary
        to have an inheritance hierarchy.
        '''
        if hasattr(self, 'trt_tensor'):
            return True
        else:
            return False

    def __hash__(self):
        if self.is_trt_wrapper():