Skip to content

GitLab

"test/srt/test_piecewise_cuda_graph.py" did not exist on "15ddd84322d431912b522ab0aed0b91056d4ef1c"
Switch branch/tag
Find file BlameHistoryPermalink
  • Matt's avatar
    Better TF docstring types (#23477) · f8b25744
    Matt authored 
    * Rework TF type hints to use | None instead of Optional[] for tf.Tensor

* Rework TF type hints to use | None instead of Optional[] for tf.Tensor

* Don't forget the imports

* Add the imports to tests too

* make fixup

* Refactor tests that depended on get_type_hints

* Better test refactor

* Fix an old hidden bug in the test_keras_fit input creation code

* Fix for the Deit tests
    f8b25744
test_modeling_tf_resnet.py 9.27 KB