update nnvm version (#3290)

apache · Sep 13, 2016 · 3dad068 · 3dad068
1 parent cf2d9cd
commit 3dad068
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/nnvm b/nnvm
diff --git a/src/executor/graph_executor.cc b/src/executor/graph_executor.cc
@@ -330,7 +330,7 @@ Graph GraphExecutor::InitGraph(nnvm::Symbol symbol,
   // other initializations
   g = nnvm::pass::InferShape(g, arg_shapes, "__shape__");
   g = nnvm::pass::InferType(g, arg_types);
-  g = nnvm::ApplyPass(g, {"PlanMemory"});
+  g = nnvm::ApplyPass(g, "PlanMemory");
   return g;
 }
 

diff --git a/src/nnvm/legacy_op_util.cc b/src/nnvm/legacy_op_util.cc
@@ -199,7 +199,7 @@ inline std::vector<NodeEntry> OpPropGradient(
   gnode->attrs = ptr->attrs;
   gnode->attrs.op = back_op;
   gnode->attrs.name = ptr->attrs.name + "_backward";
-  std::vector<NodeEntry> in_grad(prop.inputs.size());
+  std::vector<NodeEntry> in_grad(prop.arguments.size());
   for (uint32_t i = 0; i < prop.arguments.size(); ++i) {
     in_grad[i] = NodeEntry{gnode, i, 0};
   }
+4 −0		.gitignore
+5 −5		include/nnvm/c_api.h
+3 −3		include/nnvm/graph.h
+5 −5		include/nnvm/graph_attr_types.h
+33 −18		include/nnvm/pass.h
+40 −29		include/nnvm/pass_functions.h
+61 −44		include/nnvm/symbolic.h
+4 −4		include/nnvm/tuple.h
+1 −1		python/nnvm/graph.py
+5 −5		src/c_api/c_api_graph.cc
+2 −2		src/core/pass.cc
+2 −0		src/pass/gradient.cc
+26 −19		src/pass/infer_shape_type.cc
+4 −1		src/pass/plan_memory.cc