Skip to content

Load TfLite models correctly for s390x#2200

Open
yasiribmcon wants to merge 2 commits intotensorflow:masterfrom
linux-on-ibm-z:TFS_Load_TfLite_BE
Open

Load TfLite models correctly for s390x#2200
yasiribmcon wants to merge 2 commits intotensorflow:masterfrom
linux-on-ibm-z:TFS_Load_TfLite_BE

Conversation

@yasiribmcon
Copy link

Proposed changes are an extension for tensorflow/tensorflow#60190.
These changes enable string tensors in TFLite models which were generated on LE machines to be loaded successfully on BE platforms.
Below test cases from TF serving test suite are getting rectified after applying this PR on s390x(BE machines):

//tensorflow_serving/servables/tensorflow:saved_model_bundle_factory_test
//tensorflow_serving/servables/tensorflow:tflite_interpreter_pool_test
//tensorflow_serving/servables/tensorflow:tflite_session_test

These changes will not cause any regression on LE/BE platforms.

Loading TfLite model from file(little endian) correctly for s390x(big endian) systems.

Signed-off-by: Yasir Ashfaq <Yasir.Ashfaq1@ibm.com>
@yasiribmcon
Copy link
Author

Hi @gbaned

Could you please take a look at this PR when you have some time?

Thank you very much!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant