diff options
| author | Thibaut Goetghebuer-Planchon <thibaut.goetghebuer-planchon@arm.com> | 2022-11-23 11:42:43 +0000 |
|---|---|---|
| committer | Thibaut Goetghebuer-Planchon <thibaut.goetghebuer-planchon@arm.com> | 2022-12-13 10:53:38 +0000 |
| commit | 9fb745d92173abfa270e99bd5c9bd7cf85bfeb31 (patch) | |
| tree | 3a430220145a184a940131d6f7b0465dacdbde3b /tosa_checker/tosa_checker.h | |
| parent | a2bcf5f818699082adfd346eba216d96f14d6e6c (diff) | |
| download | tosa_checker-9fb745d92173abfa270e99bd5c9bd7cf85bfeb31.tar.gz | |
Models with tfl.custom ops are considered to be non TOSA compliant
Change-Id: Ib31b44b3819bbdf517b96d834879155eebb4f09c
Diffstat (limited to 'tosa_checker/tosa_checker.h')
| -rw-r--r-- | tosa_checker/tosa_checker.h | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/tosa_checker/tosa_checker.h b/tosa_checker/tosa_checker.h index d7750ea..f0e473a 100644 --- a/tosa_checker/tosa_checker.h +++ b/tosa_checker/tosa_checker.h @@ -47,6 +47,8 @@ class TOSAChecker { std::string GetMLIRTOSAModelRepresentation(bool elide_large_attrs); private: + static bool IsTOSACompatibleOp(mlir::Operation& op); + template <typename T> static std::string GetMLIRRepresentation(T&& op); |