Api add

vshampor · Apr 13, 2023 · b604b1b · b604b1b
1 parent 30eed1b
commit b604b1b
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.