pytorch
455b8739 - Introduce a match filter for SubgraphRewriter (#86430)

Commit
2 years ago
Introduce a match filter for SubgraphRewriter (#86430) This PR introduces an interface for user defined function that filters the matches in SubgraphRewriter. The function will have the following signature. callable(match: InternalMatch, original_graph: Graph, pattern_graph: Graph) -> bool This filter is applied after SubgraphMatcher returns the matches, and before replacement takes place. Pull Request resolved: https://github.com/pytorch/pytorch/pull/86430 Approved by: https://github.com/jerryzh168
Author
Committer
Parents
Loading