Skip to content

Commit 2a937d8

Browse files
committed
[fix] Add stringdoc for new param in create_executor
1 parent bf172d9 commit 2a937d8

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

python/tvm/relay/build_module.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -544,6 +544,10 @@ def create_executor(kind="debug", mod=None, device=None, target="llvm", params=N
544544
target : :py:class:`tvm.Target`
545545
The corresponding context
546546
547+
params : dict of str to NDArray
548+
Input parameters to the graph that do not change
549+
during inference time. Used for constant folding.
550+
547551
Returns
548552
-------
549553
executor : :py:class:`~tvm.relay.backend.interpreter.Executor`
@@ -555,9 +559,8 @@ def create_executor(kind="debug", mod=None, device=None, target="llvm", params=N
555559
else:
556560
device = _nd.device(str(target), 0)
557561

558-
# print("params " + params)
559562
if params is not None:
560-
mod = IRModule.from_expr(bind_params_by_name(mod["main"], params))
563+
mod = IRModule.from_expr(bind_params_by_name(mod["main"], params))
561564

562565
if isinstance(target, str):
563566
target = Target(target)

0 commit comments

Comments
 (0)