pytorch
5c396337 - Handle 1D input for xnnpack::linear (#54986)

Commit
3 years ago
Handle 1D input for xnnpack::linear (#54986) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/54986 If the input is 1D xnnpack::linear fails while aten::linear makes it (1, D) and continues Test Plan: buck test //caffe2/test:xnnpack_integration -- TestXNNPACKOps Reviewed By: kimishpatel Differential Revision: D27441966 fbshipit-source-id: dfb2c23b91247632e0e3fd2482056a503c246c39
Author
Akshit Khurana
Parents
Loading