[TensorRT] <class 'tvm.relay.op.op_attrs.PadAttrs'> has no attribute pad_value

vinx13 · October 21, 2021, 6:13pm

apache/tvm/blob/main/python/tvm/relay/op/contrib/tensorrt.py#L745


def pad_annotate_fn(expr):  # pylint: disable=unused-variable
    """Check if nn.pad is supported by TensorRT."""


    attrs, args = expr.attrs, expr.args
    if any([x.checked_type.dtype != "float32" for x in args]):
        logger.info("Only float32 inputs are supported for TensorRT.")
        return False
    if attrs.pad_mode != "constant":
        logger.info("nn.pad: pad mode is %s but must be constant.", attrs.pad_mode)
        return False
    if float(attrs.pad_value) != 0.0:
        logger.info("nn.pad: pad value is %f but must be 0.0.", float(attrs.pad_value))
        return False
    if len(attrs.pad_width) not in [4, 5]:
        logger.info("nn.pad: can only pad 4D or 5D inputs")
        return False
    if any([x != 0 for x in attrs.pad_width[0]]) or any([x != 0 for x in attrs.pad_width[1]]):
        logger.info("nn.pad: can't pad batch or channel dimensions.")
        return False
    if len(attrs.pad_width) == 5 and any([x != 0 for x in attrs.pad_width[2]]):
        logger.info("nn.pad: can only pad last two dimensions for 5D inputs.")

you need to change it to pad_value = args[1] and check isinstance(pad_value, Constant) so that you can get the numeric value of pad value