pytorch
71372b45 - [vulkan] addmm support non-vulkan inputs (#39078)

Commit
5 years ago
[vulkan] addmm support non-vulkan inputs (#39078) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/39078 Adding support of non-vulkan inputs for addmm operator: if it is not on vulkan - converting to it inside operator, if we run torchscript pretrained model - weights of linear op will be on CPU, we need this to run mobilenetV2 on Vulkan backend Test Plan: Imported from OSS Differential Revision: D21962425 Pulled By: IvanKobzarev fbshipit-source-id: 8222edd31dfb14b326d15e6fec5c8778783479df
Author
Parents
Loading