diff options
| author |  Vijay Vasudevan <vrv@google.com> | 2016-07-27 13:18:47 -0800 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2016-07-27 14:33:37 -0700 |
| commit | eaf5ad2ce467cd8730cbbab0cd96dee0d92399c1 (patch) | |
| tree | 9f5f0f44f37b624025c53fcbd5106a1731f9e934 /tensorflow/core/ops/state_ops.cc | |
| parent | 870385322db130f8d03c9205a81abec8d704d3d8 (diff) | |
TensorFlow: Annotate UnknownShape common shape function to all
ops that currently do the same in python.
Change: 128627877
Diffstat (limited to 'tensorflow/core/ops/state_ops.cc')
| -rw-r--r-- | tensorflow/core/ops/state_ops.cc | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/core/ops/state_ops.cc b/tensorflow/core/ops/state_ops.cc index a80c3a1876..6499e3bbfc 100644 --- a/tensorflow/core/ops/state_ops.cc +++ b/tensorflow/core/ops/state_ops.cc @@ -13,6 +13,7 @@ See the License for the specific language governing permissions and limitations under the License. ==============================================================================*/ +#include "tensorflow/core/framework/common_shape_fns.h" #include "tensorflow/core/framework/op.h" namespace tensorflow { @@ -24,6 +25,7 @@ REGISTER_OP("Variable") .Attr("container: string = ''") .Attr("shared_name: string = ''") .SetIsStateful() + .SetShapeFn(shape_inference::UnknownShape) .Doc(R"doc( Holds state in the form of a tensor that persists across steps. |