Skip to content

Commit 8b2cff8

Browse files
authored
[Doc] Fix minor error in doc (Add an operator to Relay) (#16282)
* Fix minor error in doc * Update relay_add_op.rst
1 parent 2da3798 commit 8b2cff8

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

docs/dev/how_to/relay_add_op.rst

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -336,10 +336,10 @@ the arguments to the call node, as below. In ``src/relay/op/tensor/transform.cc`
336336
return Call(op, {data}, Attrs(attrs), {});
337337
}
338338

339-
TVM_REGISTER_GLOBAL("relay.op._make.cumsum").set_body_typed(MakeCumprod);
339+
TVM_REGISTER_GLOBAL("relay.op._make.cumprod").set_body_typed(MakeCumprod);
340340

341341
Where ``TVM_REGISTER_GLOBAL`` exposes the ``MakeCumsum`` and ``MakeCumprod`` functions
342-
in Python via ``relay.op._make.cumsum(...)`` and ``relay.op._make.cumsum(...)``.
342+
in Python via ``relay.op._make.cumsum(...)`` and ``relay.op._make.cumprod(...)``.
343343

344344
7. Including a Cleaner Python API Hook
345345
--------------------------------------

0 commit comments

Comments
 (0)