From 946a0172f909208a5d58536c795bc73965609152 Mon Sep 17 00:00:00 2001 From: Vasily Shamporov Date: Thu, 13 Apr 2023 16:41:05 +0200 Subject: [PATCH] Api add --- nncf/onnx/engine.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/nncf/onnx/engine.py b/nncf/onnx/engine.py index 9e76d8bc3fa..586fb9874b9 100644 --- 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.