masahi commented on a change in pull request #6433:
URL: https://github.com/apache/incubator-tvm/pull/6433#discussion_r486037840
##########
File path: python/tvm/relay/frontend/pytorch.py
##########
@@ -1757,6 +1759,20 @@ def _impl(inputs, input_types):
return _impl
+def _stack(prelude):
+ def _impl(inputs, input_types):
+ if isinstance(inputs[0], list):
+ # a static python list of tensors
+ dim = inputs[1]
+ return _op.stack(inputs[0], dim)
+ else:
+ # List ADT case
+ # TODO: is there a better way to check if an input is a List ADT?
Review comment:
A proper assertion added and TODO removed
----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
For queries about this service, please contact Infrastructure at:
[email protected]
