提交 5a95c76c 编写于 作者: A A. Unique TensorFlower 提交者: TensorFlower Gardener

Set producer version in the graph used by shape refiner to run constant

folding.
Change: 150450219
上级 8e138e72
......@@ -203,6 +203,9 @@ Status ShapeRefiner::EvaluateConstantTensorForEdge(const Node* node,
bool is_constant_graph = false;
Graph subgraph(ops_registry_);
auto versions = subgraph.versions();
versions.set_producer(graph_def_version_);
subgraph.set_versions(versions);
// We identify the possibly constant subgraph to evaluate by
// recursively iterating backwards through the inputs to 'node'
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册