fa595eb8fa
Add is_tensorrt_build and tensorrt_version to build_info. PiperOrigin-RevId: 341663396 Change-Id: I239d7d52c388767932716abd501061d31324ff10
22 lines
600 B
Python
22 lines
600 B
Python
# Description:
|
|
# Contains script to generate tensorflow/python/platform/build_info.py
|
|
|
|
package(
|
|
default_visibility = ["//tensorflow:internal"],
|
|
licenses = ["notice"], # Apache 2.0
|
|
)
|
|
|
|
py_binary(
|
|
name = "gen_build_info",
|
|
srcs = ["gen_build_info.py"],
|
|
exec_compatible_with = ["@local_execution_config_platform//:platform_constraint"],
|
|
python_version = "PY3",
|
|
srcs_version = "PY2AND3",
|
|
tags = ["no-remote-exec"],
|
|
deps = [
|
|
"@local_config_cuda//cuda:cuda_config_py",
|
|
"@local_config_tensorrt//:tensorrt_config_py",
|
|
"@six_archive//:six",
|
|
],
|
|
)
|