More TF int dtype fixes #20384
Rocketknight1
merged
20 commits into
main
from
fix_test_saved_model_creation_extended
Rocketknight1
changed the title 
Add a test to ensure int dummy inputs are int64
262fe1ff
Move the test into the existing int64 test and update a lot of existi…
17ecadef
Fix remaining dummies
9dc1736a
Fix remaining dummies
29d3667b
Test for int64 serving sigs as well
ec2ba092
Update core tests to use tf.int64
39807fd3
Add better messages to the assertions
301e1b34
Update all serving sigs to int64
b5c4912c
More sneaky hiding tf.int32s
da0086cd
sgugger
approved these changes
on 2022-11-22
gante
commented
on 2022-11-22
ydshieh
approved these changes
on 2022-11-22
Add an optional int32 signature in save_pretrained
5f7d8b48
make fixup
dc5bc413
Add Amy's suggestions
f3762780
Switch all serving sigs back to tf.int32
91e15bb9
Switch all dummies to tf.int32
ef46d3e0
Adjust tests to check for tf.int32 instead of tf.int64
80f7e3be
Fix base dummy_inputs dtype
e7dc6ff5
Start casting to tf.int32 in input_processing
c9ca34c3
Change dtype for unpack_inputs test
2d8d7e5a
Add proper tf.int32 test
bc98591b
Make the alternate serving signature int64
40a3f7ab
gante
approved these changes
on 2022-11-25
Reviewers
Assignees
No one assigned
Labels
None yet
Milestone
No milestone
Login to write a write a comment.
Login via GitHub