pytorch
e332d802 - [iOS][CoreML] Remove shape information from TensorSpec (#67412)

Commit
3 years ago
[iOS][CoreML] Remove shape information from TensorSpec (#67412) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/67412 For inputs, we'll be using the shape from PyTorch tensors. For outputs, we'll be using the shape from MLMultiArray. Thus, we can decouple from the symbolic shapes defined in the compile spec. ghstack-source-id: 141746346 Test Plan: - Sandcastle - buck test pp-ios Reviewed By: hanton Differential Revision: D31299408 fbshipit-source-id: 337d5bb9efc2ff51409586c288d607399b937212
Author
Parents
Loading