Function decode

tools/Polygraphy/polygraphy/backend/trt/algorithm_selector.py:114–121

(dct)

@Decoder.register(TensorInfo)
def decode(dct):
    return TensorInfo(
        util.getattr_nested(trt, dct["tensor_format"]),
        util.getattr_nested(trt, dct["dtype"]),
        dct["strides"],
        dct["vectorized_dim"],
        dct["components_per_element"],
    )


@mod.export()