pytorch
a3670ba3 - Add option to specify custom NNAPI serializer (#61025)

Commit
3 years ago
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
Author
Akshit Khurana
Parents
Loading