pytorch
28c736a4 - Third batch of canonical aten ops (#91995)

Comment changes are shownComment changes are hidden
Commit
2 years ago
Third batch of canonical aten ops (#91995) Following aten ops appears as high frequency ops in the 14k github crawl model, and they don't have decomps: https://github.com/jansel/pytorch-jit-paritybench as_strided floor select.int topk max_pool3d_with_indices reflection_pad2d replication_pad2d replication_pad3d Full dump of aten ops from 14k model can be found here: https://docs.google.com/spreadsheets/d/1sEt0HD-0YAF5lfdOUPPZd2xIvwPL0emE7GaiqgMaTSM/edit?usp=sharing Pull Request resolved: https://github.com/pytorch/pytorch/pull/91995 Approved by: https://github.com/ezyang
Author
Committer
Parents
  • aten/src/ATen/native
    • File
      native_functions.yaml
  • torch/_inductor
    • File
      decomposition.py