From 8cce50a91ebde8343328771f6c0423c98c8167be Mon Sep 17 00:00:00 2001 From: jburnim Date: Tue, 29 Aug 2023 14:12:27 -0700 Subject: [PATCH] Update generated tensor_shape.py. PiperOrigin-RevId: 561134679 --- .../backend/numpy/gen/tensor_shape.py | 24 +++++++++++++++++-- 1 file changed, 22 insertions(+), 2 deletions(-) diff --git a/tensorflow_probability/python/internal/backend/numpy/gen/tensor_shape.py b/tensorflow_probability/python/internal/backend/numpy/gen/tensor_shape.py index dcedbd196f..e6747ca9bd 100755 --- a/tensorflow_probability/python/internal/backend/numpy/gen/tensor_shape.py +++ b/tensorflow_probability/python/internal/backend/numpy/gen/tensor_shape.py @@ -1357,8 +1357,28 @@ def most_specific_common_supertype( @doc_controls.do_not_doc_inheritable def placeholder_value(self, placeholder_context): - raise NotImplementedError("A graph placeholder is not currently supported" - "for an object of type: TensorShape.") + """See tf.types.experimental.TraceType base class.""" + return super().placeholder_value(placeholder_context) + + @doc_controls.do_not_doc_inheritable + def from_tensors(self, tensors): + """See tf.types.experimental.TraceType base class.""" + return super().from_tensors(tensors) + + @doc_controls.do_not_doc_inheritable + def to_tensors(self, value): + """See tf.types.experimental.TraceType base class.""" + return super().to_tensors(value) + + @doc_controls.do_not_doc_inheritable + def flatten(self): + """See tf.types.experimental.TraceType base class.""" + return super().flatten() + + @doc_controls.do_not_doc_inheritable + def cast(self, value, cast_context): + """See tf.types.experimental.TraceType base class.""" + return super().cast(value, cast_context) @classmethod def experimental_type_proto(cls) -> Type[tensor_shape_pb2.TensorShapeProto]: