Repository navigation
TF-TRT Backward compatibility test - #34660
tensorflow-copybara merged 11 commits into
Conversation
pooyadavoodi
commented
Nov 27, 2019
- Adds a new test to ensure TF-TRT can execute models saved in previous minor releases of TF2.x.
- The checked in model is generated using tensorflow-gpu==2.0.0
This model can be loaded and executed in tests to ensure backwards compatibility across TF versions.
|
@pooyadavoodi can you please check build failures ? |
|
I just fixed a few bugs. Any idea about these errors which seem to be related to Eigen? |
That error is hopefully fixed by this: e3a7bdb |
|
Also @sanjoy |
| from __future__ import division | ||
| from __future__ import print_function | ||
|
|
||
| import tensorflow as tf |
There was a problem hiding this comment.
There is an error when trying to submit this PR:
'import tensorflow' statements are no allowed inside TF's code base... Please import modules directly.
Also could you help to add a build rule for this file?
There was a problem hiding this comment.
This file is not supposed to be used by any test. It's supposed to be used manually by developers to generate a test model if needed. Do we still need to add it to BUILD?
There was a problem hiding this comment.
I think adding a build rule will make our life easier: we just need to run bazel run to get the model. But I don't have strong opinion. But at least replace import tensorflow with direct module imports.
There was a problem hiding this comment.
I removed the import.
I also noticed the test model was missing in the data dependency of trt_convert_test which is fixed now. I missed this because I run tests without bazel :(
I haven't added gen_tftrt_model.py to BUILD. I would need to learn how to do that first. I suggest to postpone this for now.
Also remove gen_tftrt_model.py from BUILD because this file is not supposed to be used by any test. It's supposed to be used manually by developers to generate a test model if needed.
|
Getting this now: What do you suggest? |
aaroey
left a comment
There was a problem hiding this comment.
Unapprove and try to trigger the copybara import again.
PiperOrigin-RevId: 285002333 Change-Id: I698a30181a97fb48cf3fadffefb1a7a4e046d6d4