pytorch
641f6ef8 - Implement IMethod::getArgumentNames() (#61856)

Commit
3 years ago
Implement IMethod::getArgumentNames() (#61856) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/61856 This diff did the following few things: 1. It implemented IMethod::getArgumentNames() for all IMethod's subclasses. 2. It refactors PyTorchDeployPredictor to use IMethod for model executions. Test Plan: [... ~/fbsource/fbcode/caffe2] buck test mode/dev caffe2/fb/predictor:pytorch_predictor_test -- PyTorchDeployPredictor [... ~/fbsource/fbcode/caffe2] buck test mode/dev caffe2/fb/predictor:pytorch_predictor_test -- PyTorchPredictor Reviewed By: wconstab Differential Revision: D29648756 fbshipit-source-id: e047345f26ce495a5d74d8063f7f8edc32a1b13c
Author
Jiewen Tan
Parents
Loading