Skip to content

Commit 2c14498

Browse files
committed
update doc (#47)
1 parent 689de5b commit 2c14498

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

nnvm/docs/overview.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -110,13 +110,13 @@ NNVM_REGISTER_OP(add)
110110

111111
// register to tell first input can be calculate inplace with first output
112112
NNVM_REGISTER_OP(add)
113-
.attr<FInplaceOption>("FInplaceOption", [](const NodeAttrs& attrs) {
113+
.set_attr<FInplaceOption>("FInplaceOption", [](const NodeAttrs& attrs) {
114114
return std::vector<std::pair<int, int> >{{0, 0}};
115115
});
116116

117117
NNVM_REGISTER_OP(exp)
118118
.set_num_inputs(1)
119-
.attr<FInplaceOption>("FInplaceOption", [](const NodeAttrs& attrs) {
119+
.set_attr<FInplaceOption>("FInplaceOption", [](const NodeAttrs& attrs) {
120120
return std::vector<std::pair<int, int> >{{0, 0}};
121121
});
122122
```

0 commit comments

Comments
 (0)