commit | f73d17871fc86a5f05996309cc1a781fb913ca88 | [log] [tgz] |
---|---|---|
author | Eugene Zhulenev <ezhulenev@google.com> | Fri Aug 16 16:42:47 2019 -0700 |
committer | TensorFlower Gardener <gardener@tensorflow.org> | Fri Aug 16 17:13:31 2019 -0700 |
tree | 0f411ffb453cf203f5ce91a11edc382a6bb1f384 | |
parent | 4b90e3d88a8a9e1cf9743b0987cf792a8b064518 [diff] |
[Grappler:LayoutOptimizer] Initialize fanin shape with unknown rank PiperOrigin-RevId: 263872959
diff --git a/tensorflow/core/grappler/optimizers/generic_layout_optimizer_transposer.cc b/tensorflow/core/grappler/optimizers/generic_layout_optimizer_transposer.cc index 8da1e4d..34d91a2 100644 --- a/tensorflow/core/grappler/optimizers/generic_layout_optimizer_transposer.cc +++ b/tensorflow/core/grappler/optimizers/generic_layout_optimizer_transposer.cc
@@ -443,6 +443,7 @@ string added_node_name; if (op == kOpTranspose) { TensorShapeProto input_shape_proto; + input_shape_proto.set_unknown_rank(true); if (input_shape != nullptr) { input_shape_proto = input_shape->list().shape(src_port); } else {