aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/contrib/lite/kernels/BUILD
diff options
context:
space:
mode:
authorGravatar Jared Duke <jdduke@google.com>2018-08-01 17:03:14 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2018-08-01 17:08:41 -0700
commit6e9c1b57087e15ea850b20bace7881fe95a86854 (patch)
treee816941b970b037cb59d076acd49a49d4736f321 /tensorflow/contrib/lite/kernels/BUILD
parent7dda474c941c8ec5a341dd0d82bfd5ce7c720b21 (diff)
Support TFLite framework compilation on Windows
PiperOrigin-RevId: 207020196
Diffstat (limited to 'tensorflow/contrib/lite/kernels/BUILD')
-rw-r--r--tensorflow/contrib/lite/kernels/BUILD27
1 files changed, 12 insertions, 15 deletions
diff --git a/tensorflow/contrib/lite/kernels/BUILD b/tensorflow/contrib/lite/kernels/BUILD
index 329c98f91e..40adaa4994 100644
--- a/tensorflow/contrib/lite/kernels/BUILD
+++ b/tensorflow/contrib/lite/kernels/BUILD
@@ -8,6 +8,16 @@ load("//tensorflow/contrib/lite:build_def.bzl", "tflite_copts")
load("//tensorflow/contrib/lite:special_rules.bzl", "tflite_portable_test_suite")
load("//tensorflow:tensorflow.bzl", "tf_cc_test")
+# Suppress warnings that are introduced by Eigen Tensor.
+EXTRA_EIGEN_COPTS = select({
+ "//tensorflow:ios": [
+ "-Wno-error=invalid-partial-specialization",
+ "-Wno-error=reorder",
+ ],
+ "//tensorflow:windows": [],
+ "//conditions:default": ["-Wno-error=reorder"],
+})
+
tf_cc_test(
name = "optional_tensor_test",
size = "small",
@@ -49,13 +59,7 @@ cc_library(
hdrs = [
"eigen_support.h",
],
- copts = tflite_copts() + [
- "-Wno-error=reorder",
- ] + select({
- "//tensorflow:ios": ["-Wno-error=invalid-partial-specialization"],
- "//conditions:default": [
- ],
- }),
+ copts = tflite_copts() + EXTRA_EIGEN_COPTS,
deps = [
":op_macros",
"//tensorflow/contrib/lite:arena_planner",
@@ -209,14 +213,7 @@ cc_library(
"padding.h",
"register.h",
],
- # Suppress warnings that are introduced by Eigen Tensor.
- copts = tflite_copts() + [
- "-Wno-error=reorder",
- ] + select({
- "//tensorflow:ios": ["-Wno-error=invalid-partial-specialization"],
- "//conditions:default": [
- ],
- }),
+ copts = tflite_copts() + EXTRA_EIGEN_COPTS,
deps = [
":activation_functor",
":eigen_support",