Skip to content

Commit

Permalink
[Doc] Fix minor error in doc (Add an operator to Relay) (#16282)
Browse files Browse the repository at this point in the history
* Fix minor error in doc

* Update relay_add_op.rst
  • Loading branch information
gty111 authored Jan 2, 2024
1 parent 2da3798 commit 8b2cff8
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions docs/dev/how_to/relay_add_op.rst
Original file line number Diff line number Diff line change
Expand Up @@ -336,10 +336,10 @@ the arguments to the call node, as below. In ``src/relay/op/tensor/transform.cc`
return Call(op, {data}, Attrs(attrs), {});
}

TVM_REGISTER_GLOBAL("relay.op._make.cumsum").set_body_typed(MakeCumprod);
TVM_REGISTER_GLOBAL("relay.op._make.cumprod").set_body_typed(MakeCumprod);

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

7. Including a Cleaner Python API Hook
--------------------------------------
Expand Down

0 comments on commit 8b2cff8

Please sign in to comment.