Skip to content

Conversation

duc0
Copy link
Contributor

@duc0 duc0 commented Aug 3, 2018

Summary:
SubtreeMatchCriteria now supports these additional requirements:

  • nonTerminal flag : if this is set, it means we only match the root of the subtree and do not care about the children. Example use case: to match an "input" node but does not care how the input is produced.
  • multiEdges flag : this goes together with "count" parameter and only applies when matching children nodes . If this is set, it means that the parent is expected to have multiple edges to the same subtree, rather than multiple subtrees matching the same criteria.
    Additional tests for these new logic are added to subgraph_matcher_test.cc.

Subgraph matching APIs for NNGraph is also added.

(Further enhancement to make the SubgraphMatching API constructs a Subgraph object/more diagnostic information will go later).

Differential Revision: D9156092

Summary:
Pull Request resolved: pytorch#10218

SubtreeMatchCriteria now supports:
- nonTerminal flag : if this is set, it means we only match the root of the subtree and do not care about the children. Example use case: to match an "input" node but does not care how the input is produced.
Additional tests for these new logic are added to subgraph_matcher_test.cc.

Subgraph matching APIs for NNGraph is also added.

(Further enhancement to make the SubgraphMatching API constructs a Subgraph object/more diagnostic information will go later).

Differential Revision: D9156092

fbshipit-source-id: 069f10015fa51e9a03feff75d1501c3a0210fcd8
@duc0 duc0 force-pushed the export-D9156092 branch from d99aa6d to 9816cdd Compare August 6, 2018 23:49
PenghuiCheng pushed a commit to PenghuiCheng/pytorch that referenced this pull request Aug 10, 2018
Summary:
Pull Request resolved: pytorch#10218

SubtreeMatchCriteria now supports:
- nonTerminal flag : if this is set, it means we only match the root of the subtree and do not care about the children. Example use case: to match an "input" node but does not care how the input is produced.
Additional tests for these new logic are added to subgraph_matcher_test.cc.

Subgraph matching APIs for NNGraph is also added.

(Further enhancement to make the SubgraphMatching API constructs a Subgraph object/more diagnostic information will go later).

Reviewed By: bwasti

Differential Revision: D9156092

fbshipit-source-id: 3f28ac15d9edd474b3e0cd51fd7e6f973299d061
goodlux pushed a commit to goodlux/pytorch that referenced this pull request Aug 15, 2018
Summary:
Pull Request resolved: pytorch#10218

SubtreeMatchCriteria now supports:
- nonTerminal flag : if this is set, it means we only match the root of the subtree and do not care about the children. Example use case: to match an "input" node but does not care how the input is produced.
Additional tests for these new logic are added to subgraph_matcher_test.cc.

Subgraph matching APIs for NNGraph is also added.

(Further enhancement to make the SubgraphMatching API constructs a Subgraph object/more diagnostic information will go later).

Reviewed By: bwasti

Differential Revision: D9156092

fbshipit-source-id: 3f28ac15d9edd474b3e0cd51fd7e6f973299d061
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants