diff options
author | Guangda Lai <laigd@google.com> | 2018-07-25 10:37:27 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-07-25 10:41:11 -0700 |
commit | 06b269aa33a1156b157d91c67866ec08caf9a22b (patch) | |
tree | f1fcd50328e4fc15f99c6538f2ca2d15dd72ca42 /tensorflow/contrib/BUILD | |
parent | a08e606825b3ccc220bb495f29fc9954f91cddf9 (diff) |
Reorder the dependencies to make sure trt appears before tflite, so the build
will resolve the flatbuffer symbols used by trt within the trt library.
PiperOrigin-RevId: 206016450
Diffstat (limited to 'tensorflow/contrib/BUILD')
-rw-r--r-- | tensorflow/contrib/BUILD | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/tensorflow/contrib/BUILD b/tensorflow/contrib/BUILD index a173c51879..6a4e252b44 100644 --- a/tensorflow/contrib/BUILD +++ b/tensorflow/contrib/BUILD @@ -130,8 +130,12 @@ py_library( "//tensorflow/contrib/bigtable", # depends on bigtable "//tensorflow/contrib/cloud:cloud_py", # doesn't compile on Windows "//tensorflow/contrib/ffmpeg:ffmpeg_ops_py", - "//tensorflow/contrib/lite/python:lite", # unix dependency, need to fix code + # TODO(aaroey): tensorrt dependency has to appear before tflite so the + # build can resolve its flatbuffers symbols within the tensorrt library. + # This is an issue with the tensorrt static library and will be fixed by + # the next tensorrt release, so fix the order here after that. "//tensorflow/contrib/tensorrt:init_py", # doesn't compile on windows + "//tensorflow/contrib/lite/python:lite", # unix dependency, need to fix code ]), ) |