Update compile_engine.py (#4393)

1bd5d717 · Huang, Guangtai · Tianqi Chen · 0bbdad4b · 1bd5d717
Commit 1bd5d717 authored Nov 22, 2019 by Huang, Guangtai Committed by Tianqi Chen Nov 21, 2019
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

python/tvm/relay/backend/compile_engine.py
+2 -2

No files found.
--- 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)