Akshit Khurana a3670ba377 Add option to specify custom NNAPI serializer (#61025) ... Summary: To add serializer for custom ops we can subclass default serializer and update ADDER_MAP Pull Request resolved: https://github.com/pytorch/pytorch/pull/61025 Test Plan: * pytest test/test_nnapi.py::TestNNAPI for current serializer * Custom serializers to be tested with custom ops Imported from OSS Reviewed By: anshuljain1 Differential Revision: D29480745 fbshipit-source-id: 37e3f8de3c97f6c8a486f9879ce11430ea89af34		2021-07-09 15:27:10 -07:00
..
__init__.py
prepare.py	Add option to specify custom NNAPI serializer (#61025)	2021-07-09 15:27:10 -07:00
serializer.py	Make nnapi flatten converter accept flex inputs (#61024)	2021-07-09 15:27:02 -07:00