diff options
author | Skye Wanderman-Milne <skyewm@google.com> | 2018-03-27 15:07:05 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-03-27 15:09:19 -0700 |
commit | 736e055a756cf0f99d59b67284aade01baec9799 (patch) | |
tree | 43034482ec323beb9d0289a4b5261a1510c1f3d7 /tensorflow/c/BUILD | |
parent | 6bb4f7abb03a7904fecc5b61e3ed37e9b663d6b0 (diff) |
Make _USE_C_API = True and _USE_C_SHAPES = False work with handle data, take 2.
This change makes _set_shapes_for_outputs_c_api fetch and set
Tensor._handle_data. This is necessary for running the
Python shape inference code on resource tensors.
PiperOrigin-RevId: 190681459
Diffstat (limited to 'tensorflow/c/BUILD')
-rw-r--r-- | tensorflow/c/BUILD | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/c/BUILD b/tensorflow/c/BUILD index 7f03e40d38..249135f728 100644 --- a/tensorflow/c/BUILD +++ b/tensorflow/c/BUILD @@ -283,6 +283,8 @@ tf_cuda_library( deps = [ ":c_api", ":c_api_internal", + # TODO(b/74620627): remove when _USE_C_SHAPES is removed + "//tensorflow/python:cpp_shape_inference_proto_cc", ], ) |