Class ProxyValue

exir/pass_base.py:119–178

class ProxyValue:
    # pyre-ignore
    def __init__(self, data, proxy: Union[torch.fx.Proxy, torch.fx.Node]):
        # pyre-ignore
        self.data = data
        self.proxy_or_node = proxy

    @property
    def node(self) -> torch.fx.Node:
        if isinstance(self.proxy_or_node, torch.fx.Node):
            return self.proxy_or_node
        assert isinstance(self.proxy_or_node, torch.fx.Proxy)
        return self.proxy_or_node.node

    @property
    def proxy(self) -> torch.fx.Proxy:
        if not isinstance(self.proxy_or_node, torch.fx.Proxy):
            raise RuntimeError(
                f"ProxyValue doesn't have attached Proxy object. Node: {self.proxy_or_node.format_node()}"
            )
        return self.proxy_or_node

    def to_tensor(self) -> torch.Tensor:
        assert isinstance(self.data, torch.Tensor)
        return self.data

    def is_tensor(self) -> bool:
        return isinstance(self.data, torch.Tensor)

    # pyre-ignore
    def __iter__(self):
        yield from self.data

    def __bool__(self) -> bool:
        if isinstance(self.data, (torch.SymInt, torch.SymFloat, torch.SymBool)):
            raise ExportPassBaseError(
                "ProxyValue with symbolic data cannot be used in boolean context."
            )
        return bool(self.data)

    def __int__(self):
        if isinstance(self.data, torch.SymInt):
            raise ExportPassBaseError(
                "ProxyValue with SymInt data cannot be converted to int."
            )
        return int(self.data)

    def __float__(self):
        if isinstance(self.data, torch.SymFloat):
            raise ExportPassBaseError(
                "ProxyValue with SymFloat data cannot be converted to float."
            )
        return float(self.data)

    def __index__(self):
        if isinstance(self.data, torch.SymInt):
            raise ExportPassBaseError(
                "ProxyValue with SymInt data cannot be used in index context."