diff options
author | Guangda Lai <laigd@google.com> | 2018-01-25 23:59:19 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-01-26 00:02:44 -0800 |
commit | 76f6938bafeb81a4ca41b8dac2b9c83e1286fa95 (patch) | |
tree | 75dd6285db559b19d7a0449065973780a748ef7c /tensorflow/BUILD | |
parent | ffda6079ed619df8fd3edb4db71ffc7d005c2430 (diff) |
Set up TensorRT configurations for external use, and add a test.
PiperOrigin-RevId: 183347199
Diffstat (limited to 'tensorflow/BUILD')
-rw-r--r-- | tensorflow/BUILD | 9 |
1 files changed, 9 insertions, 0 deletions
diff --git a/tensorflow/BUILD b/tensorflow/BUILD index 9099463c4f..3d2411a266 100644 --- a/tensorflow/BUILD +++ b/tensorflow/BUILD @@ -370,6 +370,14 @@ config_setting( visibility = ["//visibility:public"], ) +# TODO(laigd): consider removing this option and make TensorRT enabled +# automatically when CUDA is enabled. +config_setting( + name = "with_tensorrt_support", + values = {"define": "with_tensorrt_support=true"}, + visibility = ["//visibility:public"], +) + package_group( name = "internal", packages = [ @@ -558,6 +566,7 @@ filegroup( "//tensorflow/contrib/tensor_forest/proto:all_files", "//tensorflow/contrib/tensorboard:all_files", "//tensorflow/contrib/tensorboard/db:all_files", + "//tensorflow/contrib/tensorrt:all_files", "//tensorflow/contrib/testing:all_files", "//tensorflow/contrib/text:all_files", "//tensorflow/contrib/tfprof:all_files", |