update test (#48)

apache · May 29, 2018 · 35f54ec · 35f54ec
1 parent 2c14498
commit 35f54ec
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/nnvm/tests/cpp/op_test.cc b/nnvm/tests/cpp/op_test.cc
@@ -6,10 +6,10 @@
 NNVM_REGISTER_OP(add)
 .describe("add two data together")
 .set_num_inputs(2)
-.attr("inplace_pair", std::make_pair(0, 0));
+.set_attr("inplace_pair", std::make_pair(0, 0));
 
 NNVM_REGISTER_OP(add)
-.attr<std::string>("nick_name", "plus");
+.set_attr<std::string>("nick_name", "plus");
 
 
 TEST(Op, GetAttr) {