search

google-ai-edge / ai-edge-torch

Supporting PyTorch models with the Google AI Edge TFLite runtime.
Apache License 2.0
199 stars 23 forks source link

Add search model api #59

Closed wuruoyu closed 3 weeks ago

wuruoyu commented 3 weeks ago

Add _search_model API, which find subgraphs that satisfy the predicate defined by the users. Refactor find_culprits to invoke the _search_model.

Please find design doc at go/pytorch-search-model-api. BUG=None