Expose Normal/Ragged/Sparse KerasTensor as public API.

Although they are still implementation details, this will allow user to do type checking as well as type annotation for their code.

PiperOrigin-RevId: 516295859
This commit is contained in:
Scott Zhu 2023-03-13 12:54:40 -07:00 committed by TensorFlower Gardener
parent 8c3822edbb
commit f6aeedfa1e

View File

@ -71,6 +71,11 @@
evaluating Keras models trained with
`tf.distribute.ParameterServerStrategy`, via the
`exact_evaluation_shards` argument in `Model.fit` and `Model.evaluate`.
* Added `tf.keras.__internal__.KerasTensor`,
`tf.keras.__internal__.SparseKerasTensor`, and
`tf.keras.__internal__.RaggedKerasTensor` classes. You can use these
classes to do instance type checking and type annotations for
layer/model inputs and outputs.
* `tf.function`: