aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/contrib/hvx
diff options
context:
space:
mode:
Diffstat (limited to 'tensorflow/contrib/hvx')
-rw-r--r--tensorflow/contrib/hvx/clock_cycle_profiling/BUILD3
-rw-r--r--tensorflow/contrib/hvx/hvx_ops_support_checker/BUILD4
2 files changed, 5 insertions, 2 deletions
diff --git a/tensorflow/contrib/hvx/clock_cycle_profiling/BUILD b/tensorflow/contrib/hvx/clock_cycle_profiling/BUILD
index 621466c9ed..8c92e33bdf 100644
--- a/tensorflow/contrib/hvx/clock_cycle_profiling/BUILD
+++ b/tensorflow/contrib/hvx/clock_cycle_profiling/BUILD
@@ -6,6 +6,7 @@ licenses(["notice"]) # Apache 2.0
load(
"//tensorflow:tensorflow.bzl",
"tf_copts",
+ "tf_cc_binary",
)
exports_files(["LICENSE"])
@@ -24,7 +25,7 @@ filegroup(
visibility = ["//tensorflow:__subpackages__"],
)
-cc_binary(
+tf_cc_binary(
name = "clock_cycle_profiling",
testonly = 1,
srcs = ["clock_cycle_profiling_main.cc"],
diff --git a/tensorflow/contrib/hvx/hvx_ops_support_checker/BUILD b/tensorflow/contrib/hvx/hvx_ops_support_checker/BUILD
index 3497c84d58..909dc396a3 100644
--- a/tensorflow/contrib/hvx/hvx_ops_support_checker/BUILD
+++ b/tensorflow/contrib/hvx/hvx_ops_support_checker/BUILD
@@ -8,6 +8,8 @@ licenses(["notice"]) # Apache 2.0
exports_files(["LICENSE"])
+load("//tensorflow:tensorflow.bzl", "tf_cc_binary")
+
filegroup(
name = "all_files",
srcs = glob(
@@ -19,7 +21,7 @@ filegroup(
),
)
-cc_binary(
+tf_cc_binary(
name = "hvx_ops_support_checker",
testonly = 1,
srcs = ["hvx_ops_support_checker_main.cc"],