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	Add OPT (#17088)	2022-05-12 12:24:35 +02:00
test_modeling_flax_opt.py	Add final_layer_norm to OPT model (#17785)	2022-06-21 20:26:36 +02:00
test_modeling_opt.py	Adding OPTForSeqClassification class (#18123)	2022-07-20 10:14:21 +02:00
test_modeling_tf_opt.py	Update serving code to enable saved_model=True (#18153)	2022-07-22 18:05:38 +01:00