Function lookup

tests/unittest/trt/python_plugin/test_plugin_wrapper.py:62–64

(x, y)

                   dtype=torch_dtype_to_trt(dtype))

        def lookup(x, y):
            lookup_plugin = LookUpPlugin(to_torch, fp32_output)
            return lookup_plugin(x, y)

        output = lookup(x, y)