amyeroberts 8e8384663d Update serving code to enable saved_model=True (#18153) ... * Add serving_output and serving methods to some vision models * Add serving outputs for DeiT * Don't convert hidden states - differing shapes * Make saveable * Fix up * Make swin saveable * Add in tests * Fix funnel tests (can't convert to tensor) * Fix numpy call * Tidy up a bit * Add in hidden states - resnet * Remove numpy * Fix failing tests - tensor shape and skipping tests * Remove duplicated function * PR comments - formatting and var names * PR comments Add suggestions made by Joao Gante: * Use tf.shape instead of shape_list * Use @tooslow decorator on tests * Simplify some of the logic * PR comments Address Yih-Dar Sheih comments - making tensor names consistent and make types float * Types consistent with docs; disable test on swin (slow) * CI trigger * Change input_features to float32 * Add serving_output for segformer * Fixup Co-authored-by: Amy Roberts <amyeroberts@users.noreply.github.com>		2022-07-22 18:05:38 +01:00
..
__init__.py	Move test model folders (#17034)	2022-05-03 14:42:02 +02:00
test_modeling_lxmert.py	Fx support for Deberta-v[1-2], Hubert and LXMERT (#17539)	2022-06-07 18:05:20 +02:00
test_modeling_tf_lxmert.py	Update serving code to enable saved_model=True (#18153)	2022-07-22 18:05:38 +01:00
test_tokenization_lxmert.py	Move test model folders (#17034)	2022-05-03 14:42:02 +02:00