| commit | 06e2d90cc1905713147f05c65651170944ae18c0 | [log] [tgz] |
|---|---|---|
| author | Summer Yue <summeryue@google.com> | Fri Feb 05 14:11:13 2021 -0800 |
| committer | TensorFlower Gardener <gardener@tensorflow.org> | Fri Feb 05 14:15:35 2021 -0800 |
| tree | 1635ed7281a1c467703018e6e630619ba5af884c | |
| parent | 46bd5ec5f8bb24257d3e503b5d16e36f787d38d9 [diff] |
Register TF BoundedTensorSpec for json serialization. PiperOrigin-RevId: 355925369 Change-Id: I5a9916dc4b9c88873a386f69e91be6828fe4b4ea
diff --git a/tensorflow/python/framework/tensor_spec.py b/tensorflow/python/framework/tensor_spec.py index 68c2320..04ceff8 100644 --- a/tensorflow/python/framework/tensor_spec.py +++ b/tensorflow/python/framework/tensor_spec.py
@@ -220,6 +220,7 @@ # TODO(b/133606651): Should is_compatible_with should check min/max bounds? +@type_spec.register("tf.BoundedTensorSpec") class BoundedTensorSpec(TensorSpec): """A `TensorSpec` that specifies minimum and maximum values.