experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fix issues with the tests and add mistakenly taken out macro back

Browse Source
This commit is contained in:
Sami Kama 2018-02-02 16:02:31 -08:00
parent 45fcb8f2d3
commit e0be56b9cc
5 changed files with 8 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tensorflow/contrib/cmake/python_modules.txt
View File

@ -407,6 +407,7 @@ tensorflow/contrib/tensorboard/plugins
tensorflow/contrib/tensorboard/plugins/projector tensorflow/contrib/tensorboard/plugins/projector
# TODO(sami): Add cmake implementations # TODO(sami): Add cmake implementations
# tensorflow/contrib/tensorrt/python # tensorflow/contrib/tensorrt/python
# tensorflow/contrib/tensorrt/python/ops
tensorflow/contrib/tensor_forest tensorflow/contrib/tensor_forest
tensorflow/contrib/tensor_forest/client tensorflow/contrib/tensor_forest/client
tensorflow/contrib/tensor_forest/hybrid tensorflow/contrib/tensor_forest/hybrid

4
tensorflow/contrib/tensorrt/BUILD
View File

@ -78,10 +78,10 @@ cc_library(
copts = tf_copts(), copts = tf_copts(),
deps = [ deps = [
":trt_logging", ":trt_logging",
"//tensorflow/core:stream_executor_headers_lib",
"//tensorflow/core:framework_headers_lib", "//tensorflow/core:framework_headers_lib",
"//tensorflow/core:gpu_headers_lib", "//tensorflow/core:gpu_headers_lib",
"//tensorflow/core:lib_proto_parsing", "//tensorflow/core:lib_proto_parsing",
"//tensorflow/core:stream_executor_headers_lib",
"//third_party/eigen3", "//third_party/eigen3",
"@local_config_tensorrt//:nv_infer", "@local_config_tensorrt//:nv_infer",
"@nsync//:nsync_headers", "@nsync//:nsync_headers",
@ -186,9 +186,9 @@ tf_cuda_library(
deps = [ deps = [
":segment", ":segment",
":trt_logging", ":trt_logging",
"//tensorflow/core:graph",
"//tensorflow/core:framework_headers_lib", "//tensorflow/core:framework_headers_lib",
"//tensorflow/core:framework_lite", "//tensorflow/core:framework_lite",
"//tensorflow/core:graph",
"//tensorflow/core:protos_all_cc", "//tensorflow/core:protos_all_cc",
"//tensorflow/core/grappler:devices", "//tensorflow/core/grappler:devices",
"//tensorflow/core/grappler/clusters:virtual_cluster", "//tensorflow/core/grappler/clusters:virtual_cluster",

5
tensorflow/tools/pip_package/BUILD
View File

@ -182,8 +182,9 @@ sh_binary(
"//tensorflow/python:test_ops", "//tensorflow/python:test_ops",
"//tensorflow/tools/dist_test/server:grpc_tensorflow_server", "//tensorflow/tools/dist_test/server:grpc_tensorflow_server",
], ],
}) + if_mkl(["//third_party/mkl:intel_binary_blob"]) }) + if_mkl(["//third_party/mkl:intel_binary_blob"]) + if_tensorrt([
+ if_tensorrt(["//tensorflow/contrib/tensorrt:init_py"]), "//tensorflow/contrib/tensorrt:init_py",
]),
) )
# A genrule for generating a marker file for the pip package on Windows # A genrule for generating a marker file for the pip package on Windows

1
third_party/tensorrt/BUILD.tpl vendored
View File

@ -1,6 +1,7 @@
# NVIDIA TensorRT # NVIDIA TensorRT
# A high-performance deep learning inference optimizer and runtime. # A high-performance deep learning inference optimizer and runtime.
load("@local_config_cuda//cuda:build_defs.bzl", "cuda_default_copts")
licenses(["notice"]) licenses(["notice"])
exports_files(["LICENSE"]) exports_files(["LICENSE"])