Update compile_engine.py (#4393)

diff --git a/python/tvm/relay/backend/compile_engine.py b/python/tvm/relay/backend/compile_engine.py
index 545f96a..6c690a9 100644 (file)
--- a/python/tvm/relay/backend/compile_engine.py
+++ b/python/tvm/relay/backend/compile_engine.py
@@ -116,8 +116,8 @@ class CompileEngine(NodeBase):
 
         Returns
         -------
-        cached_func: CachedFunc
-            The result of lowering.
+        jited_func: tvm.Function
+            The result of jited function.
         """
         key = _get_cache_key(source_func, target)
         return _backend._CompileEngineJIT(self, key)