Skip to content

T
tic
Commit 6e08e0e8 by Eric Junyuan Xie Committed by Tianqi Chen
Options

fix error message (#90) 

* fix error message

* fix
parent 2cb9b90c
Showing with 1 additions and 1 deletions
nnvm/src/pass/infer_shape_type.cc
......@@ -136,7 +136,7 @@ Graph InferAttr(Graph &&ret,
try {
forward_known = finfer(inode.source->attrs, &ishape, &oshape);
} catch (const std::exception& e) {
throw dmlc::Error(e.what() + std::string(" with ") + inode.source->attrs.name);
throw dmlc::Error("Error in operator " + inode.source->attrs.name + ": " + e.what());
}
} else {
CHECK(!last_iter)
......
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