Commit e91897e2 authored by Jayaram Bobba's avatar Jayaram Bobba Committed by Robert Kimball

Emit cacheable attribute on parameter node (#1240)

parent 4334fe56
......@@ -1202,6 +1202,7 @@ static json write(const Node& n, bool binary_constant_data)
{
auto tmp = dynamic_cast<const op::Parameter*>(&n);
node["shape"] = tmp->get_shape();
node["cacheable"] = tmp->get_cacheable();
node["element_type"] = write_element_type(tmp->get_element_type());
}
else if (node_op == "Product")
......
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