Method add_tensor

tensorrt_llm/runtime/generation.py:2204–2205

(x, name)

            })

        def add_tensor(x, name):
            return tensors.update({name: sym(x, name)})

        def add_tensor_with_shape(x, name, shape):
            return tensors.update(