diff options
author | Guangda Lai <laigd@google.com> | 2018-09-13 10:44:21 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-13 10:48:07 -0700 |
commit | 685f2832daa7084cd1bf484e8a7bb4333e246428 (patch) | |
tree | 36c06f0b15235f4abd57f282dafa45ee97ad5182 /tensorflow/contrib/BUILD | |
parent | 1050e5dc93cd579607495df6086f3cec2d9aa1f4 (diff) |
Add TF-TRT kernels/ops to contrib_kernels and contrib_ops_op_lib, so TF serving
can use them.
PiperOrigin-RevId: 212838380
Diffstat (limited to 'tensorflow/contrib/BUILD')
-rw-r--r-- | tensorflow/contrib/BUILD | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/tensorflow/contrib/BUILD b/tensorflow/contrib/BUILD index 798f499870..d98a24994c 100644 --- a/tensorflow/contrib/BUILD +++ b/tensorflow/contrib/BUILD @@ -166,7 +166,9 @@ cc_library( "//tensorflow/contrib/kinesis:dataset_kernels", ], "//conditions:default": [], - }), + }) + if_not_windows([ + "//tensorflow/contrib/tensorrt:trt_engine_op_kernel", + ]), ) cc_library( @@ -203,5 +205,7 @@ cc_library( "//tensorflow/contrib/kinesis:dataset_ops_op_lib", ], "//conditions:default": [], - }), + }) + if_not_windows([ + "//tensorflow/contrib/tensorrt:trt_engine_op_op_lib", + ]), ) |