Class _DLTensorDTypeProxy

python/tvm/relax/expr.py:313–390

A proxy object for unpacking DLDatatype from DLTensor Exposes accessors for `DLDataType` fields `type_code`, `lanes`, and `bits` within a `DLTensor::dtype`. Accessing these fields will produce `relax.Call` expressions, representing the field's runtime value. If the datatype of the

class _DLTensorDTypeProxy(tvm.runtime.ObjectConvertible):
    """A proxy object for unpacking DLDatatype from DLTensor

    Exposes accessors for `DLDataType` fields `type_code`, `lanes`,
    and `bits` within a `DLTensor::dtype`.  Accessing these fields
    will produce `relax.Call` expressions, representing the field's
    runtime value.  If the datatype of the tensor is known at
    compile-time, the `relax.Call` will be normalized into a
    `relax.PrimValue`, with no runtime cost.

    Parameters
    ----------
    tensor: relax.Expr

        The relax tensor (or a variable referring to a relax tensor),
        whose runtime shape is being inspected.

    """

    def __init__(self, tensor):
        self.tensor = tensor

    def asobject(self):
        """Provide expected in error message

        This method is called when `_DLTensorDTypeProxy` is used in a
        context that requires a `relax.Expr`.  This usage is not
        supported, and raising an error here can provide suggested
        fixes that are not present in the default error message from
        `tvm.runtime.convert`.
        """

        fields = [f"{self.tensor}.dtype.{field}" for field in ["type_code", "bits", "lanes"]]
        raise TypeError(
            f"{self.tensor}.dtype cannot be converted to a relax expression, "
            f"and should be used as a proxy object to access "
            f"fields {fields}"
        )

    @property
    def type_code(self) -> Expr:
        """Accessor for the DLDataType::bits field

        Returns
        -------
        type_code: Expr

            The type code of the DLTensor.  See the `DLDeviceType`
            enum in `dlpack.h` for more information.
        """
        op = tvm.ir.Op.get("relax.inspect.tensor_dtype_code")
        return tvm.relax.Call(op, [self.tensor])

    @property
    def lanes(self) -> Expr:
        """Accessor for the DLDataType::bits field

        Returns