diff options
| author |  Manjunath Kudlur <keveman@google.com> | 2016-12-07 10:38:21 -0800 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2016-12-07 10:43:35 -0800 |
| commit | a79a7a21358c12d1b9096b357ea5267fc7800775 (patch) | |
| tree | bb90586b1a5a009c196d7ec512372a68c626d9fd /tensorflow/cc/BUILD | |
| parent | c8cab5483f1bd6efd70dadba6f7e6433d1c4207d (diff) | |
Added a fused instance normalization op for quantized input.
- Eigen and Neon intrinsics based implementations.
- Added test suitable for regressions and standalone runs on Android.
- Made it possible to compile the C++ API on android platform, so it's easy
to write tests.
Change: 141326710
Diffstat (limited to 'tensorflow/cc/BUILD')
| -rw-r--r-- | tensorflow/cc/BUILD | 41 |
1 files changed, 30 insertions, 11 deletions
diff --git a/tensorflow/cc/BUILD b/tensorflow/cc/BUILD index b2a074120d..d3a16c57f6 100644 --- a/tensorflow/cc/BUILD +++ b/tensorflow/cc/BUILD @@ -10,9 +10,13 @@ licenses(["notice"]) # Apache 2.0 exports_files(["LICENSE"]) -load("//tensorflow:tensorflow.bzl", "tf_cc_test") -load("//tensorflow:tensorflow.bzl", "tf_copts") -load("//tensorflow:tensorflow.bzl", "tf_gen_op_wrappers_cc") +load( + "//tensorflow:tensorflow.bzl", + "tf_cc_test", + "tf_copts", + "tf_gen_op_wrappers_cc", + "cc_library_with_android_deps", +) cc_library( name = "gradients", @@ -104,10 +108,11 @@ cc_library( ], ) -cc_library( +cc_library_with_android_deps( name = "ops", srcs = ["framework/ops.cc"], hdrs = ["framework/ops.h"], + android_deps = ["//tensorflow/core:android_tensorflow_lib"], deps = [ "//tensorflow/core:core_cpu", "//tensorflow/core:framework", @@ -117,12 +122,15 @@ cc_library( ], ) -cc_library( +cc_library_with_android_deps( name = "scope", srcs = ["framework/scope.cc"], hdrs = ["framework/scope.h"], - deps = [ + android_deps = ["//tensorflow/core:android_tensorflow_lib"], + common_deps = [ ":ops", + ], + deps = [ "//tensorflow/core:core_cpu", "//tensorflow/core:framework", "//tensorflow/core:lib", @@ -143,13 +151,16 @@ tf_cc_test( ], ) -cc_library( +cc_library_with_android_deps( name = "client_session", srcs = ["client/client_session.cc"], hdrs = ["client/client_session.h"], - deps = [ + android_deps = ["//tensorflow/core:android_tensorflow_lib"], + common_deps = [ ":ops", ":scope", + ], + deps = [ "//tensorflow/core:core_cpu", "//tensorflow/core:lib", "//tensorflow/core:protos_all_cc", @@ -173,13 +184,18 @@ tf_cc_test( ], ) -cc_library( +cc_library_with_android_deps( name = "const_op", srcs = ["ops/const_op.cc"], hdrs = ["ops/const_op.h"], - deps = [ + android_deps = [ + "//tensorflow/core:android_tensorflow_lib", + ], + common_deps = [ ":ops", ":scope", + ], + deps = [ "//tensorflow/core:core_cpu", "//tensorflow/core:framework", ], @@ -354,13 +370,16 @@ tf_gen_op_wrappers_cc( visibility = ["//tensorflow:internal"], ) -cc_library( +cc_library_with_android_deps( name = "cc_op_gen_main", srcs = [ "framework/cc_op_gen.cc", "framework/cc_op_gen.h", "framework/cc_op_gen_main.cc", ], + android_deps = [ + "//tensorflow/core:android_tensorflow_lib", + ], copts = tf_copts(), deps = [ "//tensorflow/core:framework", |