diff options
| author |  Saurabh Saxena <srbs@google.com> | 2018-10-02 13:18:27 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-10-02 13:22:56 -0700 |
| commit | 8d12c635cc48e896da0bcac1cd568bd6381ca64e (patch) | |
| tree | d651bbcfdd325e649c230c19424acc62c28de725 /tensorflow/core/grappler | |
| parent | 78e4ce52aeda5a10ddaf5e64ea8958f439a2f9f2 (diff) | |
Support shape_invariants in while_v2. Note that this arg is temporary and may be replaced by automatic shape inference in TF 2.0 (or before).
Add a output_shapes attr to While op to allow output shapes to be different from the incoming loop_vars.
PiperOrigin-RevId: 215446737
Diffstat (limited to 'tensorflow/core/grappler')
| -rw-r--r-- | tensorflow/core/grappler/costs/graph_properties_testdata/function_functional_while.pbtxt | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/tensorflow/core/grappler/costs/graph_properties_testdata/function_functional_while.pbtxt b/tensorflow/core/grappler/costs/graph_properties_testdata/function_functional_while.pbtxt index c94ee2f227..0ec95dd684 100644 --- a/tensorflow/core/grappler/costs/graph_properties_testdata/function_functional_while.pbtxt +++ b/tensorflow/core/grappler/costs/graph_properties_testdata/function_functional_while.pbtxt @@ -88,6 +88,13 @@ library { } } } + attr { + key: "output_shapes" + value { + list { + } + } + } } ret { key: "while" |