masahi commented on a change in pull request #5306: [Torch] Support Python list, more realistic recurrent networks URL: https://github.com/apache/incubator-tvm/pull/5306#discussion_r407273733
########## File path: python/tvm/relay/frontend/pytorch.py ########## @@ -25,20 +25,97 @@ import numpy as np import tvm -from tvm.ir import module as _module from .. import analysis as _analysis from .. import expr as _expr from .. import op as _op +from ..function import Function +from .. import transform +from ..ty import TupleType, TensorType, Any from ..loops import while_loop from .common import get_relay_op from .common import infer_shape as _infer_shape from .common import infer_value as _infer_value +from ..prelude import Prelude, StaticTensorArrayOps, get_tensor_array_shape from . import qnn_torch __all__ = ["from_pytorch"] + +# List ADT utilities +def _infer_type_with_prelude(val, prelude): Review comment: Thanks for pointing this out. I added this when your TF frontend PR hasn't been merged. Now we can simplify this function using infer_type in common.py ---------------------------------------------------------------- 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] With regards, Apache Git Services
