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 /third_party/tensorrt/build_defs.bzl.tpl | |
| parent | ffda6079ed619df8fd3edb4db71ffc7d005c2430 (diff) | |
Set up TensorRT configurations for external use, and add a test.
PiperOrigin-RevId: 183347199
Diffstat (limited to 'third_party/tensorrt/build_defs.bzl.tpl')
| -rw-r--r-- | third_party/tensorrt/build_defs.bzl.tpl | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/third_party/tensorrt/build_defs.bzl.tpl b/third_party/tensorrt/build_defs.bzl.tpl new file mode 100644 index 0000000000..0dc3a7ba2d --- /dev/null +++ b/third_party/tensorrt/build_defs.bzl.tpl @@ -0,0 +1,7 @@ +# Build configurations for TensorRT. + +def if_tensorrt(if_true, if_false=[]): + """Tests whether TensorRT was enabled during the configure process.""" + if %{tensorrt_is_configured}: + return if_true + return if_false |