[GitHub] [tvm] LeshengJin commented on a diff in pull request #15670: [Unity][Frontend][NN] Enable tuple/list input

LeshengJin commented on code in PR #15670: URL: https://github.com/apache/tvm/pull/15670#discussion_r1318047379 ## python/tvm/relax/frontend/nn/spec.py: ## @@ -59,6 +61,25 @@ def __repr__(self) -> str: return f"Tensor([{shape}], '{self.dtype}')" +class Tuple: +

[GitHub] [tvm] LeshengJin commented on a diff in pull request #15670: [Unity][Frontend][NN] Enable tuple/list input

LeshengJin commented on code in PR #15670: URL: https://github.com/apache/tvm/pull/15670#discussion_r1318045540 ## python/tvm/relax/frontend/nn/spec.py: ## @@ -59,6 +59,32 @@ def __repr__(self) -> str: return f"Tensor([{shape}], '{self.dtype}')" +class TupleList: +

[GitHub] [tvm] LeshengJin commented on a diff in pull request #15670: [Unity][Frontend][NN] Enable tuple/list input

LeshengJin commented on code in PR #15670: URL: https://github.com/apache/tvm/pull/15670#discussion_r1318045404 ## python/tvm/relax/frontend/nn/spec.py: ## @@ -59,6 +59,32 @@ def __repr__(self) -> str: return f"Tensor([{shape}], '{self.dtype}')" +class TupleList: +

[GitHub] [tvm] LeshengJin commented on a diff in pull request #15670: [Unity][Frontend][NN] Enable tuple/list input

LeshengJin commented on code in PR #15670: URL: https://github.com/apache/tvm/pull/15670#discussion_r1318044932 ## python/tvm/relax/frontend/nn/core.py: ## @@ -411,6 +412,10 @@ def jit( # pylint: disable=too-many-arguments # Compile mod and feed it to VM