search

apple / coremltools

Core ML tools contain supporting tools for Core ML model conversion, editing, and validation.
https://coremltools.readme.io
BSD 3-Clause "New" or "Revised" License
4.46k stars 647 forks source link

[Torch.Export] Support Sort Ops Translation #2399

Closed YifanShenSZ closed 5 days ago

YifanShenSZ commented 5 days ago

Support sort ops translation from torch.export

  1. sort
  2. argsort

Testing: ✅ https://gitlab.com/coremltools1/coremltools/-/commit/16a7f14d1339eae93c168b1fb1eedaa97bc08e61/pipelines