pytorch/torch/backends
Akshit Khurana d0fd41dcfe Add size op in nnapi serializer (#52026)
Summary:
serializer didn't support aten::size

Pull Request resolved: https://github.com/pytorch/pytorch/pull/52026

Test Plan: Torchvision Mobilenetv2 [script](https://pytorch.org/tutorials/prototype/nnapi_mobilenetv2.html) works. [Test](ecfed07cc5) to be merged after [this PR](https://github.com/pytorch/pytorch/pull/47521/files) is merged

Reviewed By: dreiss

Differential Revision: D26363133

Pulled By: axitkhurana

fbshipit-source-id: 772a6bea62bca69f8bba19c25c582a1734a70eb1
2021-02-10 15:57:01 -08:00
..
_nnapi Add size op in nnapi serializer (#52026) 2021-02-10 15:57:01 -08:00
cuda [reland] Enable TF32 support for cuBLAS (#41498) 2020-07-15 21:00:55 -07:00
cudnn Mention TF32 on related docs (#44690) 2020-09-16 19:18:30 -07:00
mkl
mkldnn Add torch.backends.mkldnn.enabled flag (#25459) 2019-09-11 12:09:40 -07:00
openmp
quantized Enable typechecking for torch.testing._internal.common_quantized.* (#44805) 2020-09-17 14:24:32 -07:00
xnnpack Remove py2 compatible future imports (#44735) 2020-09-16 12:55:57 -07:00
__init__.py Add torch.backends.mkldnn.enabled flag (#25459) 2019-09-11 12:09:40 -07:00