Skip to content

T
tic
Commit c30ef503 by Tianqi Chen
Options

update doc (#47)

parent 6ebbea5d
Showing with 2 additions and 2 deletions
nnvm/docs/overview.md
...@@ -110,13 +110,13 @@ NNVM_REGISTER_OP(add) ...@@ -110,13 +110,13 @@ NNVM_REGISTER_OP(add)
// register to tell first input can be calculate inplace with first output // register to tell first input can be calculate inplace with first output
NNVM_REGISTER_OP(add) NNVM_REGISTER_OP(add)
.attr<FInplaceOption>("FInplaceOption", [](const NodeAttrs& attrs) { .set_attr<FInplaceOption>("FInplaceOption", [](const NodeAttrs& attrs) {
return std::vector<std::pair<int, int> >{{0, 0}}; return std::vector<std::pair<int, int> >{{0, 0}};
}); });
NNVM_REGISTER_OP(exp) NNVM_REGISTER_OP(exp)
.set_num_inputs(1) .set_num_inputs(1)
.attr<FInplaceOption>("FInplaceOption", [](const NodeAttrs& attrs) { .set_attr<FInplaceOption>("FInplaceOption", [](const NodeAttrs& attrs) {
return std::vector<std::pair<int, int> >{{0, 0}}; return std::vector<std::pair<int, int> >{{0, 0}};
}); });
``` ```
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment