kevinthesun 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_r407268130
########## 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: Can we reuse infer_type defined in common.py. The only difference is that we need to get checked_type for the return of infer_type. ---------------------------------------------------------------- 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
