pytorch
7dd5da20 - JIT pass to insert XNNPACK ops (#34048)

Commit
5 years ago
JIT pass to insert XNNPACK ops (#34048) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/34048 Rewrites the graph to insert xnnpack prepack and packed run ops for conv2d and linear. Test Plan: python test/test_xnnpack_integration.py Imported from OSS Differential Revision: D20185658 fbshipit-source-id: c4c073c912ad33e822e7beb4ed86c9f895129d55
Author
Parents
Loading