pytorch
4c30fc72 - Integrate XNNPACK with custom class for packing weights. (#34047)

Commit
4 years ago
Integrate XNNPACK with custom class for packing weights. (#34047) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/34047 This PR integrates the added xnnpack conv2d and linear op via custom class registration for packed weights. The packed struct is serializable. Test Plan: python test test/test_xnnpack_integration.py Imported from OSS Differential Revision: D20185657 fbshipit-source-id: fc7e692d8f913e493b293b02d92f4e78536d7698
Author
Parents
Loading