Skip to content

T
tic
Commit 70163c6b by Hu Shiwen Committed by Tianqi Chen
Options

error info (#69)

parent e9b9206f
Showing with 5 additions and 1 deletions
nnvm/src/pass/infer_shape_type.cc
......@@ -119,7 +119,11 @@ Graph InferAttr(Graph &&ret,
<< "Attribute " << infer_name
<< " is not registed by op " << inode.source->op()->name;
// Call inference function of the operator.
forward_known = finfer(inode.source->attrs, &ishape, &oshape);
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);
}
}
// Save to the result map.
for (uint32_t i = 0; i < num_inputs; ++i) {
......
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