pytorch
9d9ce19d - [split cat fx passes] Normalize squeeze (#102294)

Commit
1 year ago
[split cat fx passes] Normalize squeeze (#102294) Summary: Sometimes, squeeze can be a "call_method" instead of a "call_function". Normalizing it will make it amenable to pattern matching by passes like "split->squeeze" Test Plan: * CI tests Differential Revision: D46031846 Pull Request resolved: https://github.com/pytorch/pytorch/pull/102294 Approved by: https://github.com/jansel
Committer
Parents
Loading