aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/contrib/quantization
diff options
context:
space:
mode:
authorGravatar Gunhan Gulsoy <gunan@google.com>2016-09-09 12:39:39 -0800
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2016-09-09 13:49:17 -0700
commit3127edd15972b8850845015d4eca51882e5350da (patch)
treea3e952897d3fbc8bad90b707f12753d5f62d8d85 /tensorflow/contrib/quantization
parent6187789bfd8c8e037480333aa50db0ed9cb0d000 (diff)
To prepare for removal of cuda_cc_test(s) build rule, make all build rules
consistent by making all rules take srcs arg. Change: 132710890
Diffstat (limited to 'tensorflow/contrib/quantization')
-rw-r--r--tensorflow/contrib/quantization/kernels/BUILD10
-rw-r--r--tensorflow/contrib/quantization/kernels/hexagon/BUILD1
2 files changed, 11 insertions, 0 deletions
diff --git a/tensorflow/contrib/quantization/kernels/BUILD b/tensorflow/contrib/quantization/kernels/BUILD
index 6621c450fc..6be2ccaa07 100644
--- a/tensorflow/contrib/quantization/kernels/BUILD
+++ b/tensorflow/contrib/quantization/kernels/BUILD
@@ -118,6 +118,7 @@ py_library(
tf_cc_test(
name = "quantize_down_and_shrink_range_op_test",
size = "small",
+ srcs = ["quantize_down_and_shrink_range_op_test.cc"],
deps = [
":quantized_ops",
"//tensorflow/contrib/quantization:cc_array_ops",
@@ -135,6 +136,7 @@ tf_cc_test(
tf_cc_test(
name = "quantization_utils_test",
+ srcs = ["quantization_utils_test.cc"],
deps = [
":quantized_ops",
"//tensorflow/contrib/quantization:cc_array_ops",
@@ -154,6 +156,7 @@ tf_cc_test(
tf_cc_test(
name = "quantized_activation_ops_test",
+ srcs = ["quantized_activation_ops_test.cc"],
deps = [
":quantized_ops",
"//tensorflow/contrib/quantization:cc_array_ops",
@@ -172,6 +175,7 @@ tf_cc_test(
tf_cc_test(
name = "quantized_bias_add_op_test",
size = "small",
+ srcs = ["quantized_bias_add_op_test.cc"],
deps = [
":quantized_ops",
"//tensorflow/contrib/quantization:cc_array_ops",
@@ -190,6 +194,7 @@ tf_cc_test(
tf_cc_test(
name = "quantized_conv_ops_test",
size = "small",
+ srcs = ["quantized_conv_ops_test.cc"],
deps = [
":quantized_ops",
"//tensorflow/contrib/quantization:cc_array_ops",
@@ -208,6 +213,7 @@ tf_cc_test(
tf_cc_test(
name = "quantize_op_test",
size = "small",
+ srcs = ["quantize_op_test.cc"],
deps = [
":quantized_ops",
"//tensorflow/contrib/quantization:cc_array_ops",
@@ -226,6 +232,7 @@ tf_cc_test(
tf_cc_test(
name = "quantized_matmul_op_test",
size = "small",
+ srcs = ["quantized_matmul_op_test.cc"],
deps = [
":quantized_ops",
"//tensorflow/contrib/quantization:cc_array_ops",
@@ -244,6 +251,7 @@ tf_cc_test(
tf_cc_test(
name = "quantized_pooling_ops_test",
size = "small",
+ srcs = ["quantized_pooling_ops_test.cc"],
deps = [
":quantized_ops",
"//tensorflow/contrib/quantization:cc_array_ops",
@@ -262,6 +270,7 @@ tf_cc_test(
tf_cc_test(
name = "quantized_concat_op_test",
size = "small",
+ srcs = ["quantized_concat_op_test.cc"],
deps = [
":quantized_ops",
"//tensorflow/contrib/quantization:cc_array_ops",
@@ -282,6 +291,7 @@ tf_cc_test(
tf_cc_test(
name = "quantized_batch_norm_op_test",
size = "small",
+ srcs = ["quantized_batch_norm_op_test.cc"],
deps = [
":quantized_ops",
"//tensorflow/contrib/quantization:cc_array_ops",
diff --git a/tensorflow/contrib/quantization/kernels/hexagon/BUILD b/tensorflow/contrib/quantization/kernels/hexagon/BUILD
index 3aebd36e2f..b57a2ac1b5 100644
--- a/tensorflow/contrib/quantization/kernels/hexagon/BUILD
+++ b/tensorflow/contrib/quantization/kernels/hexagon/BUILD
@@ -28,6 +28,7 @@ filegroup(
tf_cc_test(
name = "quantized_matmul_op_for_hexagon_test",
size = "small",
+ srcs = ["quantized_matmul_op_for_hexagon_test.cc"],
deps = [
"//tensorflow/contrib/quantization:cc_array_ops",
"//tensorflow/contrib/quantization:cc_math_ops",