diff options
author | gracehoney <31743510+aaroey@users.noreply.github.com> | 2018-06-19 12:39:24 -0700 |
---|---|---|
committer | gracehoney <31743510+aaroey@users.noreply.github.com> | 2018-06-19 12:39:24 -0700 |
commit | 520384df634f64cb6d803884f5f0c9462a6ef9fd (patch) | |
tree | 8757ac249a491e45355a20608550bf471a57e29f /tensorflow/contrib/tensorrt/BUILD | |
parent | f3f6ef4c74982f867bf0d1e96f79097598f55eb3 (diff) |
Use TrtUniquePtrType for all builder/network/engine construction; add build rules for utils.h; add more TODOs
Diffstat (limited to 'tensorflow/contrib/tensorrt/BUILD')
-rw-r--r-- | tensorflow/contrib/tensorrt/BUILD | 9 |
1 files changed, 8 insertions, 1 deletions
diff --git a/tensorflow/contrib/tensorrt/BUILD b/tensorflow/contrib/tensorrt/BUILD index fd0f97f3af..e7b3fe38e5 100644 --- a/tensorflow/contrib/tensorrt/BUILD +++ b/tensorflow/contrib/tensorrt/BUILD @@ -87,6 +87,7 @@ cc_library( ":trt_plugins", ":trt_resources", ":trt_conversion", + ":utils", "//tensorflow/core:gpu_headers_lib", "//tensorflow/core:lib_proto_parsing", "//tensorflow/core:stream_executor_headers_lib", @@ -94,7 +95,7 @@ cc_library( ] + if_tensorrt([ "@local_config_tensorrt//:nv_infer", ]) + tf_custom_op_library_additional_deps(), - # TODO(laigd) + # TODO(laigd): fix this by merging header file in cc file. alwayslink = 1, # buildozer: disable=alwayslink-with-hdrs ) @@ -232,6 +233,7 @@ tf_cuda_library( ":trt_plugins", ":trt_logging", ":trt_resources", + ":utils", "//tensorflow/core/grappler/clusters:cluster", "//tensorflow/core/grappler/optimizers:custom_graph_optimizer", "//tensorflow/core/grappler/optimizers:custom_graph_optimizer_registry", @@ -337,3 +339,8 @@ py_test( "//tensorflow/python:framework_test_lib", ], ) + +cc_library( + name = "utils", + hdrs = ["convert/utils.h"], +) |