Api add

vshampor · Apr 14, 2023 · f3c67db · f3c67db
1 parent ef2d46c
commit f3c67db
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/nncf/onnx/engine.py b/nncf/onnx/engine.py
@@ -17,7 +17,9 @@
 import onnxruntime as rt
 
 from nncf.common.engine import Engine
+from nncf.common.api_marker import api
 
+@api()
 class ONNXEngine(Engine):
     """
     Engine for ONNX backend using ONNXRuntime to infer the model.