diff --git a/nnvm/docs/overview.md b/nnvm/docs/overview.md index 064519057661..e4c9fe4b1641 100644 --- a/nnvm/docs/overview.md +++ b/nnvm/docs/overview.md @@ -110,13 +110,13 @@ NNVM_REGISTER_OP(add) // register to tell first input can be calculate inplace with first output NNVM_REGISTER_OP(add) -.attr("FInplaceOption", [](const NodeAttrs& attrs) { +.set_attr("FInplaceOption", [](const NodeAttrs& attrs) { return std::vector >{{0, 0}}; }); NNVM_REGISTER_OP(exp) .set_num_inputs(1) -.attr("FInplaceOption", [](const NodeAttrs& attrs) { +.set_attr("FInplaceOption", [](const NodeAttrs& attrs) { return std::vector >{{0, 0}}; }); ```