Commit 5b9bff4a authored by Peter Hawkins's avatar Peter Hawkins Committed by TensorFlower Gardener
Browse files

[TF:XLA] Use xla::Shape instead of TensorShape for representing... 

[TF:XLA] Use xla::Shape instead of TensorShape for representing argument/resource update shapes to XlaCompiler.

Change in preparation for adding implementations of the Stack... operators in the Tensorflow/XLA bridge. The representation of a Stack will be a tuple of (stack value, stack size). xla::Shape can represent tuples, but TensorShape cannot.

PiperOrigin-RevId: 161391774
parent bf267f36
Show whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment