ShapeRefiner fix: some variant-type tensors have handle data.
authorSkye Wanderman-Milne <skyewm@google.com>
Mon, 7 May 2018 23:16:32 +0000 (16:16 -0700)
committerTensorFlower Gardener <gardener@tensorflow.org>
Tue, 8 May 2018 00:51:42 +0000 (17:51 -0700)
commit6e1784b6b4e0542de0ac3ebd790633c6db9cfe46
tree5ffcd6ef0fc625869f8c41921a96243659bb17fb
parent5802096c267c805f6a69798aac10aefef759bb9f
ShapeRefiner fix: some variant-type tensors have handle data.

ShapeRefiner::AddNode() would only propagate handle data for
DT_RESOURCE tensors, but not DT_VARIANT. The Python shape inference
logic in common_shapes.py handled this correct, which is why we didn't
notice this earlier. In particular, list ops use DT_VARIANT with
handle data.
PiperOrigin-RevId: 195739586
tensorflow/core/common_runtime/shape_refiner.cc
tensorflow/python/kernel_tests/list_ops_test.py