jjohnson-arm opened a new pull request #5204: [Frontend][Torch] Fix up graph input handling URL: https://github.com/apache/incubator-tvm/pull/5204 Thanks for contributing to TVM! Please refer to guideline https://tvm.apache.org/docs/contribute/ for useful information and tips. After the pull request is submitted, please request code reviews from [Reviewers](https://github.com/apache/incubator-tvm/blob/master/CONTRIBUTORS.md#reviewers) by @ them in the pull request thread. From: https://discuss.tvm.ai/t/pytorch-frontend-graph-input-names-can-change-using-loaded-torchscript/6055 Split as two commits to make it easier to review: * Simplify operator input handling * remove outputs and output_index_map and extend use of input_vars * Allow user supplied input names to override graph inputs * PyTorch inputs now expected as list rather than dictionary - [(name, shape), (name, shape)...] * Input names given with from_pytorch() are now set as the graph input names and should be used in set_input() Review request: @masahi
---------------------------------------------------------------- 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
