tree b76301bf97efe91c2970b1079740a1ff4d3821b7
parent 1b55b855a5589122e2cbb489e2aadf9600ee5618
author Derek Murray <mrry@google.com> 1566840059 -0700
committer TensorFlower Gardener <gardener@tensorflow.org> 1566840819 -0700

Optimize TF_Output construction when building `tf.Tensor` objects.

Each `tf.Tensor` caches a SWIG-wrapped `TF_Output` object to map that tensor to the C API's graph. We currently lazily cache this object in `tf.Tensor` on its first consumption, but it turns out that we already build the object for all tensors as a by-product of calculating tensor types. This change saves the original object in each `tf.Tensor` when it is constructed, which decreases the overhead of graph construction.

PiperOrigin-RevId: 265484399