diff options
| author |  Benoit Steiner <bsteiner@google.com> | 2017-11-28 12:49:38 -0800 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2017-11-28 12:53:20 -0800 |
| commit | a86f2d2c1af7ac0e5c36eedb18d74c022737fc25 (patch) | |
| tree | 22e3ca5caba030b4b73e7fdec3040315c4138fce /tensorflow/core/grappler/costs/virtual_scheduler.cc | |
| parent | b911049edfbb4a4eb07b3b46ed144da6cd33f9c1 (diff) | |
Added an option to assume that the shape of fed nodes in unknown since any shape can be actually used.
PiperOrigin-RevId: 177203023
Diffstat (limited to 'tensorflow/core/grappler/costs/virtual_scheduler.cc')
| -rw-r--r-- | tensorflow/core/grappler/costs/virtual_scheduler.cc | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/core/grappler/costs/virtual_scheduler.cc b/tensorflow/core/grappler/costs/virtual_scheduler.cc index e5e1ee3292..6640de668d 100644 --- a/tensorflow/core/grappler/costs/virtual_scheduler.cc +++ b/tensorflow/core/grappler/costs/virtual_scheduler.cc @@ -122,7 +122,7 @@ Status VirtualScheduler::Init() { // Construct graph properties. Status status; if (use_static_shapes_) { - status = graph_properties_.InferStatically(); + status = graph_properties_.InferStatically(true); } else { status = graph_properties_.InferDynamically(cluster_); } |