pytorch/test/cpp
Jacob Szwejbka 1c5f63d86d [Pytorch Edge] Model Ops compatibility api (#57501)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/57501

Add an api _get_model_ops_and_info to get root operators and versioning info of a model in both cxx and python, and the input can be from a file path or buffer.
ghstack-source-id: 129620112

Test Plan: unit test.

Reviewed By: xcheng16, raziel

Differential Revision: D28162765

fbshipit-source-id: 4413c1e906b8a872e4a717d849da37347adbbea4
2021-05-24 12:00:06 -07:00
..
api Add inference mode python bindings and tests (#58045) 2021-05-13 08:55:35 -07:00
common
dist_autograd Fix distributed autograd gradients synchronization (#57792) 2021-05-09 17:32:59 -07:00
jit [Pytorch Edge] Model Ops compatibility api (#57501) 2021-05-24 12:00:06 -07:00
lite_interpreter_runtime [Pytorch Backend delegation] Add api for backend lowering to query debug (#55462) 2021-05-22 08:33:07 -07:00
rpc Revert D28474878: Always use intrusive_ptr for Message (1 out of 2) 2021-05-21 19:24:22 -07:00
tensorexpr [NNC] Do not optimize conditionals when the corresponding loop is not normalized (#57675) 2021-05-18 14:25:53 -07:00
__init__.py