Fetch C shapes for ops created by import_graph_def with C API enabled.
authorSkye Wanderman-Milne <skyewm@google.com>
Thu, 22 Mar 2018 21:09:59 +0000 (14:09 -0700)
committerTensorFlower Gardener <gardener@tensorflow.org>
Thu, 22 Mar 2018 21:12:36 +0000 (14:12 -0700)
commit48b0fb7a524425d57547dc23093d869538b888db
treef1664d09b2950eec59614c324c0142e2d7d5da7f
parent804f98e5bc0a75284f5f92569e5c82fe88b455ad
Fetch C shapes for ops created by import_graph_def with C API enabled.

If _USE_C_API = True, this change makes us always fetch shapes using
the C API after calling TF_ImportGraphDef, even if _USE_C_SHAPES =
False. This is necessary to preserve the shapes specified by the
"_output_shapes" attr on imported NodeDefs (note that this attr isn't
present on the NodeDefs of the imported nodes, so there's no other way
to recover this information after calling TF_ImportGraphDef).

PiperOrigin-RevId: 190122991
tensorflow/python/framework/ops.py