tests/test_modeling_tf_common.py · fc1d841b2db7d110fe5983b3a00c80508edf79eb · chenpangpang / transformers · GitLab

Find file Blame History Permalink

change shape to support dynamic batch input in tf.function XLA generate for tf serving (#18372) · fc1d841b
nlpcat authored Aug 04, 2022
```
* change shape to support dynamic batch input in tf.generate

* add tests
Co-authored-by: nlpcatcode <nlpcodecat@gmail.com>
```
fc1d841b

test_modeling_tf_common.py 106 KB